Update flipstreamviewer.py

fix to clean warning
This commit is contained in:
sakura1bgx
2025-01-30 20:09:20 +09:00
committed by GitHub
parent 30eed43b99
commit 76480e9198
+1 -1
View File
@@ -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,
)