Spaces:

MD1998
/

Native_Bot

Runtime error

MD1998 commited on Apr 14, 2024

Commit

090154d

verified ·

1 Parent(s): d0a604a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ pipeline=transformers.pipeline(
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,
     device_map="auto",
-    max_length=500,
     do_sample=True,
     top_k=10,
     num_return_sequences=1,

     torch_dtype=torch.bfloat16,
     trust_remote_code=True,
     device_map="auto",
+    max_length=64,
     do_sample=True,
     top_k=10,
     num_return_sequences=1,