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