Spaces:
Paused
Paused
zhanghaoji
commited on
Commit
•
e6edca8
1
Parent(s):
0471713
fix tensor
Browse files
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 |
-
|
76 |
-
return
|
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):
|