Update app.py
Browse files
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)
|