Flash-VStream-demo

Paused

zhanghaoji commited on Jun 26

Commit

5d42cba

•

1 Parent(s): 6b323f3

fix demo

Files changed (1) hide show

flash_vstream/serve/demo.py CHANGED Viewed

@@ -99,7 +99,6 @@ class Chat:
                 images=images_tensor,
                 do_sample=True,
                 temperature=temperature,
-                num_beams=1,
                 max_new_tokens=max_new_tokens,
                 use_cache=True,
                 stopping_criteria=[stopping_criteria])

                 images=images_tensor,
                 do_sample=True,
                 temperature=temperature,
                 max_new_tokens=max_new_tokens,
                 use_cache=True,
                 stopping_criteria=[stopping_criteria])