Spaces:

blockenters
/

chat

Sleeping

blockenters commited on Jan 28

Commit

a545b17

1 Parent(s): d29211b

add

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,11 +57,10 @@ def main():
         # 모델 생성
         response = pipe(
             prompt,
-            max_length=2048,
             temperature=0.7,
             do_sample=True,
-            truncation=True,
-            max_new_tokens=512
         )
         bot_text = response[0]["generated_text"]

         # 모델 생성
         response = pipe(
             prompt,
+            max_new_tokens=512,
             temperature=0.7,
             do_sample=True,
+            truncation=True
         )
         bot_text = response[0]["generated_text"]