Update README.md
Browse files
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
|