Hugging Face's logo

Spaces:
snap-research
/
weights2weights
Running on Zero

App Files Community
5
amildravid4292 commited on
Commit
54b4d47
·
verified ·
1 Parent(s): c79af8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ df = torch.load(f"{models_path}/files/identity_df.pt")
46
  weight_dimensions = torch.load(f"{models_path}/files/weight_dimensions.pt")
47
  pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt").bfloat16().to(device)
48
 
49
- @spaces.GPU()
50
  unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
51
 
52
  @spaces.GPU()
 
46
  weight_dimensions = torch.load(f"{models_path}/files/weight_dimensions.pt")
47
  pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt").bfloat16().to(device)
48
 
49
+
50
  unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
51
 
52
  @spaces.GPU()