Muhammad Taqi Raza
commited on
Commit
·
7d2ae5b
1
Parent(s):
fed9194
modifying requirements.txt
Browse files
inference/cli_demo_camera_i2v_pcd.py
CHANGED
|
@@ -376,6 +376,8 @@ def generate_video(
|
|
| 376 |
]
|
| 377 |
|
| 378 |
latents = torch.stack(latents) # [B, T, C, H, W]
|
|
|
|
|
|
|
| 379 |
|
| 380 |
print(f"Type of latents: {type(latents)}")
|
| 381 |
print(f"Length of latents: {len(latents)}")
|
|
|
|
| 376 |
]
|
| 377 |
|
| 378 |
latents = torch.stack(latents) # [B, T, C, H, W]
|
| 379 |
+
latents = latents.to(device)
|
| 380 |
+
|
| 381 |
|
| 382 |
print(f"Type of latents: {type(latents)}")
|
| 383 |
print(f"Length of latents: {len(latents)}")
|