From 76480e91989f3003dae93ffe021c66c72be1f4e2 Mon Sep 17 00:00:00 2001 From: sakura1bgx Date: Thu, 30 Jan 2025 20:09:20 +0900 Subject: [PATCH] Update flipstreamviewer.py fix to clean warning --- flipstreamviewer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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, )