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):
|