Mizuiro-sakura
/

open-calm-large-finetuned-databricks-dolly

Text Generation

text-generation-inference

Model card Files Files and versions Community

Mizuiro-sakura commited on Jun 28, 2023

Commit

5d508ca

•

1 Parent(s): a8d2016

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -79,7 +79,7 @@ def generate(instruction,input=None,maxTokens=256):
     # 推論
     prompt = generate_prompt({'instruction':instruction,'input':input})
     input_ids = tokenizer(prompt, return_tensors="pt", truncation=True).input_ids
-    outputs = model.generate(
         input_ids=input_ids,
         max_new_tokens=maxTokens,
         do_sample=True,

     # 推論
     prompt = generate_prompt({'instruction':instruction,'input':input})
     input_ids = tokenizer(prompt, return_tensors="pt", truncation=True).input_ids
+    outputs = model.generate(
         input_ids=input_ids,
         max_new_tokens=maxTokens,
         do_sample=True,