Update flipstreamviewer.py
fix to clean warning
This commit is contained in:
+1
-1
@@ -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,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user