Spaces:

ecyht2
/

transformers-nlp-template

Sleeping

ecyht2 commited on Jan 22, 2024

Commit

ecae120

verified ·

1 Parent(s): a19677d

fix: Fixed crash with model kwargs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,8 +48,7 @@ def generate(
         max_new_tokens=max_new_tokens,
         top_p=top_p,
         repetition_penalty=repetition_penalty,
-        do_sample=True,
-        seed=42,
     )
     logging.debug("Generate Args: %s", generate_kwargs)

         max_new_tokens=max_new_tokens,
         top_p=top_p,
         repetition_penalty=repetition_penalty,
+        do_sample=True
     )
     logging.debug("Generate Args: %s", generate_kwargs)