Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,7 @@ tags:
|
|
| 9 |
base_model: baichuan-inc/Baichuan2-7B-Chat
|
| 10 |
metrics:
|
| 11 |
- bleu
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
|
|
@@ -37,4 +38,4 @@ pipeline = transformers.pipeline(
|
|
| 37 |
|
| 38 |
outputs = pipeline(prompt, max_new_tokens=512, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
|
| 39 |
print(outputs[0]["generated_text"])
|
| 40 |
-
```
|
|
|
|
| 9 |
base_model: baichuan-inc/Baichuan2-7B-Chat
|
| 10 |
metrics:
|
| 11 |
- bleu
|
| 12 |
+
pipeline_tag: text-generation
|
| 13 |
---
|
| 14 |
|
| 15 |
|
|
|
|
| 38 |
|
| 39 |
outputs = pipeline(prompt, max_new_tokens=512, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
|
| 40 |
print(outputs[0]["generated_text"])
|
| 41 |
+
```
|