Spaces:

k3ybladewielder
/

cloudwalk_swarm

Sleeping

k3ybladewielder commited on 22 days ago

Commit

4520f8d

verified ·

1 Parent(s): 95a4f54

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -80,7 +80,7 @@ def get_llm():
         "text-generation",
         model=model,
         tokenizer=tokenizer,
-        max_new_tokens=550,
         temperature=0.6,
         #eos_token_id=tokenizer.eos_token_id,
         eos_token_id=tokenizer.convert_tokens_to_ids("</s>"),

         "text-generation",
         model=model,
         tokenizer=tokenizer,
+        max_new_tokens=250,
         temperature=0.6,
         #eos_token_id=tokenizer.eos_token_id,
         eos_token_id=tokenizer.convert_tokens_to_ids("</s>"),