OpenLab-NLP
/

model-prototype

Model card Files Files and versions

Yuchan commited on Nov 23

Commit

5d17913

·

verified ·

1 Parent(s): 0e72098

Update Inference.py

Files changed (1) hide show

Inference.py +1 -1

Inference.py CHANGED Viewed

@@ -226,4 +226,4 @@ def generate_text_topp(model, prompt, max_len=512, max_gen=512, p=0.9, temperatu
     return ids_to_text(generated)
 print("\n\n===== 생성 결과 =====")
-print(generate_text_topp(model, "지난 2년 동안", p=0.9))

     return ids_to_text(generated)
 print("\n\n===== 생성 결과 =====")
+print(generate_text_topp(model, "지난 2년 동안", p=0.8))