Hugging Face's logo

Spaces:
heheyas
/
V3D
Running on Zero

App Files Community
2
heheyas commited on
Commit
f8a13ca
•
1 Parent(s): 21f4df4

update app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def do_sample(
139
  return video_path
140
 
141
 
142
- device = torch.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")
 
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")