diff --git a/flipstreamviewer.py b/flipstreamviewer.py index 08bfdaa..b706f4a 100644 --- a/flipstreamviewer.py +++ b/flipstreamviewer.py @@ -1856,7 +1856,7 @@ class FlipStreamSegMask: input_ids=inputs["input_ids"].cuda(), pixel_values=inputs["pixel_values"].cuda(), max_new_tokens=1024, - early_stopping=True, + early_stopping=False, do_sample=False, num_beams=1, )