Longz1003 commited on
Commit
cca934f
·
verified ·
1 Parent(s): 6afcb57

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -1
README.md CHANGED
@@ -58,4 +58,18 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
58
  # 测试模型
59
  inputs = tokenizer("你好,今天的天气怎么样?", return_tensors="pt")
60
  outputs = model.generate(**inputs)
61
- print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  # 测试模型
59
  inputs = tokenizer("你好,今天的天气怎么样?", return_tensors="pt")
60
  outputs = model.generate(**inputs)
61
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
62
+
63
+ torch>=1.10.0
64
+ transformers>=4.30.0
65
+ accelerate>=0.18.0
66
+ datasets>=2.0.0
67
+ scipy>=1.7.0
68
+ sentencepiece>=0.1.96
69
+ pytorch-lightning>=1.5.0
70
+ pyyaml>=5.4.1
71
+ tqdm>=4.62.0
72
+ requests>=2.26.0
73
+ huggingface_hub>=0.12.0
74
+ tensorboard>=2.6.0
75
+ scikit-learn>=0.24.2