Spaces:

HuggingFaceM4
/

ai_raven

Runtime error

VictorSanh commited on Mar 10, 2024

Commit

65966af

1 Parent(s): 6893a8c

fix length

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,7 @@ def model_inference(
     generation_kwargs = dict(
         inputs,
         bad_words_ids=BAD_WORDS_IDS,
-        max_length=4,
     )
     # Regular generation version
     generated_ids = MODEL.generate(**generation_kwargs)

     generation_kwargs = dict(
         inputs,
         bad_words_ids=BAD_WORDS_IDS,
+        max_length=128,
     )
     # Regular generation version
     generated_ids = MODEL.generate(**generation_kwargs)