Mizuiro-sakura
commited on
Commit
•
5d508ca
1
Parent(s):
a8d2016
Update README.md
Browse files
README.md
CHANGED
@@ -79,7 +79,7 @@ def generate(instruction,input=None,maxTokens=256):
|
|
79 |
# 推論
|
80 |
prompt = generate_prompt({'instruction':instruction,'input':input})
|
81 |
input_ids = tokenizer(prompt, return_tensors="pt", truncation=True).input_ids
|
82 |
-
outputs = model.generate(
|
83 |
input_ids=input_ids,
|
84 |
max_new_tokens=maxTokens,
|
85 |
do_sample=True,
|
|
|
79 |
# 推論
|
80 |
prompt = generate_prompt({'instruction':instruction,'input':input})
|
81 |
input_ids = tokenizer(prompt, return_tensors="pt", truncation=True).input_ids
|
82 |
+
outputs = model.generate(
|
83 |
input_ids=input_ids,
|
84 |
max_new_tokens=maxTokens,
|
85 |
do_sample=True,
|