amildravid4292 commited on
Commit
af1e565
·
verified ·
1 Parent(s): 6630510

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt").bfloat16().to(d
49
 
50
  unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
51
 
52
- @spaces.GPU()
53
  def sample_model():
54
  global unet
55
  del unet
 
49
 
50
  unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
51
 
52
+
53
  def sample_model():
54
  global unet
55
  del unet