Hugging Face's logo

Spaces:
snap-research
/
weights2weights
Running on Zero

App Files Community
5
amildravid4292 commited on
Commit
82639a6
·
verified ·
1 Parent(s): ae6aebd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ device = "cuda:0"
27
  generator = torch.Generator(device=device)
28
 
29
 
30
- models_path = snapshot_download(repo_id="Snapchat/w2w")
31
 
32
  mean = torch.load(f"{models_path}/mean.pt").bfloat16().to(device)
33
  std = torch.load(f"{models_path}/std.pt").bfloat16().to(device)
 
27
  generator = torch.Generator(device=device)
28
 
29
 
30
+ models_path = snapshot_download(repo_id="Snapchat/w2w/files")
31
 
32
  mean = torch.load(f"{models_path}/mean.pt").bfloat16().to(device)
33
  std = torch.load(f"{models_path}/std.pt").bfloat16().to(device)