homework1

Sleeping

EdmundYi commited on Jun 27

Commit

a93d250

verified ·

1 Parent(s): f1d9d07

Update model_wrappers/model_a.py

Files changed (1) hide show

model_wrappers/model_a.py CHANGED Viewed

@@ -6,7 +6,7 @@ model_a = pipeline("text-generation", model="IDEA-CCNL/Wenzhong-GPT2-110M", toke
 def run_model_a(prompt: str) -> str:
     output = model_a(prompt,
-                     max_length=100,
                      do_sample=True,
                      temperature=0.8,
                      top_k=50,

 def run_model_a(prompt: str) -> str:
     output = model_a(prompt,
+                     max_length=200,
                      do_sample=True,
                      temperature=0.8,
                      top_k=50,