Hugging Face's logo

Spaces:
snap-research
/
weights2weights
Running on Zero

App Files Community
5
multimodalart HF staff commited on
Commit
05fbd23
·
verified ·
1 Parent(s): 01fbaff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ global text_encoder
31
  global tokenizer
32
  global noise_scheduler
33
  global network
34
- device = "cuda:0"
35
  #generator = torch.Generator(device=device)
36
 
37
  models_path = snapshot_download(repo_id="Snapchat/w2w")
 
31
  global tokenizer
32
  global noise_scheduler
33
  global network
34
+ device = "cuda"
35
  #generator = torch.Generator(device=device)
36
 
37
  models_path = snapshot_download(repo_id="Snapchat/w2w")