zhanghaoji commited on
Commit
e6edca8
1 Parent(s): 0471713

fix tensor

Browse files
Files changed (1) hide show
  1. flash_vstream/serve/demo.py +2 -2
flash_vstream/serve/demo.py CHANGED
@@ -72,8 +72,8 @@ class Chat:
72
  sample_pos = all_pos
73
 
74
  patch_numpy = vreader.get_batch(sample_pos).asnumpy()
75
- patch_images = Image.fromarray(patch_numpy)
76
- return patch_images
77
 
78
  @torch.inference_mode()
79
  def generate(self, images_tensor, prompt, first_run, state):
 
72
  sample_pos = all_pos
73
 
74
  patch_numpy = vreader.get_batch(sample_pos).asnumpy()
75
+ print("patch_numpy", patch_numpy.shape)
76
+ return patch_numpy
77
 
78
  @torch.inference_mode()
79
  def generate(self, images_tensor, prompt, first_run, state):