JiunYi commited on
Commit
87d1b59
1 Parent(s): 8de8e6b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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
+ ```