Laurie commited on
Commit
64045f9
1 Parent(s): 18f4d61

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -1
README.md CHANGED
@@ -1,3 +1,13 @@
 
 
 
 
 
 
 
 
 
 
1
  使用transformers推理:
2
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
@@ -13,4 +23,4 @@
13
 
14
  inputs = tokenizer(["<human>:{}\n<bot>:".format(query)], return_tensors="pt")
15
  inputs = inputs.to("cuda")
16
- generate_ids = model.generate(**inputs, max_new_tokens=256, streamer=streamer)
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets:
4
+ - Laurie/alpaca_chinese_dataset
5
+ language:
6
+ - zh
7
+ - en
8
+ library_name: transformers
9
+ pipeline_tag: text2text-generation
10
+ ---
11
  使用transformers推理:
12
 
13
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
 
23
 
24
  inputs = tokenizer(["<human>:{}\n<bot>:".format(query)], return_tensors="pt")
25
  inputs = inputs.to("cuda")
26
+ generate_ids = model.generate(**inputs, max_new_tokens=256, streamer=streamer)