Spaces:

autonomous019
/

image_story_generator

Runtime error

autonomous019 commited on Aug 31, 2022

Commit

9163025

•

1 Parent(s): e2924dd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ def create_story(text_seed):
         MinLengthLogitsProcessor(10, eos_token_id=model.config.eos_token_id),
     ]
   )
-  stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=200)])
   outputs = model.greedy_search(
     input_ids, logits_processor=logits_processor, stopping_criteria=stopping_criteria

         MinLengthLogitsProcessor(10, eos_token_id=model.config.eos_token_id),
     ]
   )
+  stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=100)])
   outputs = model.greedy_search(
     input_ids, logits_processor=logits_processor, stopping_criteria=stopping_criteria