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

fix tensor

Browse files
Files changed (1) hide show
  1. flash_vstream/serve/demo.py +2 -1
flash_vstream/serve/demo.py CHANGED
@@ -71,7 +71,8 @@ class Chat:
71
  else:
72
  sample_pos = all_pos
73
 
74
- patch_images = [Image.fromarray(f) for f in vreader.get_batch(sample_pos).asnumpy()]
 
75
  return patch_images
76
 
77
  @torch.inference_mode()
 
71
  else:
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()