Mizuiro-sakura commited on
Commit
5d508ca
1 Parent(s): a8d2016

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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,