Tommi09 commited on
Commit
cd87ffb
·
verified ·
1 Parent(s): 7854c13

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -0
README.md ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model:
4
+ - deepseek-ai/DeepSeek-R1-7b-base
5
+ datasets:
6
+ - FreedomIntelligence/Huatuo26M-Lite
7
+ ---
8
+ ## 使用示例
9
+ ```python
10
+ from transformers import AutoModelForCausalLM, AutoTokenizer
11
+
12
+ model = AutoModelForCausalLM.from_pretrained("Tommi09/MedicalChatBot-7b-test")
13
+ tokenizer = AutoTokenizer.from_pretrained("Tommi09/MedicalChatBot-7b-test")
14
+
15
+ inputs = tokenizer("气血两虚, 有哪些症状?", return_tensors="pt")
16
+ outputs = model.generate(**inputs)
17
+ print(tokenizer.decode(outputs[0]))