Spaces:
Runtime error
Runtime error
heheyas
commited on
Commit
·
f8a13ca
1
Parent(s):
21f4df4
update app
Browse files
app.py
CHANGED
|
@@ -139,7 +139,7 @@ def do_sample(
|
|
| 139 |
return video_path
|
| 140 |
|
| 141 |
|
| 142 |
-
device =
|
| 143 |
|
| 144 |
# download
|
| 145 |
V3D_ckpt_path = hf_hub_download(repo_id="heheyas/V3D", filename="V3D.ckpt")
|
|
|
|
| 139 |
return video_path
|
| 140 |
|
| 141 |
|
| 142 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 143 |
|
| 144 |
# download
|
| 145 |
V3D_ckpt_path = hf_hub_download(repo_id="heheyas/V3D", filename="V3D.ckpt")
|