amildravid4292 commited on
Commit
fef9fbe
·
verified ·
1 Parent(s): f8040f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -242,6 +242,7 @@ class main():
242
  @torch.no_grad()
243
  @spaces.GPU(duration=120)
244
  def edit_inference(self, prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
 
245
  device = self.device
246
  self.unet.to(device)
247
  self.text_encoder.to(device)
 
242
  @torch.no_grad()
243
  @spaces.GPU(duration=120)
244
  def edit_inference(self, prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
245
+ print("start")
246
  device = self.device
247
  self.unet.to(device)
248
  self.text_encoder.to(device)