Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -114,7 +114,6 @@ class ModelManager:
|
|
| 114 |
self.pipe.unet.set_attn_processor(AttnProcessor2_0())
|
| 115 |
self.pipe.vae.set_attn_processor(AttnProcessor2_0())
|
| 116 |
if USE_TORCH_COMPILE:
|
| 117 |
-
progress(0.5, desc="Compiling the model...")
|
| 118 |
self.pipe.unet = torch.compile(self.pipe.unet, mode="reduce-overhead", fullgraph=True)
|
| 119 |
|
| 120 |
if self.realesrgan_x2 is None:
|
|
|
|
| 114 |
self.pipe.unet.set_attn_processor(AttnProcessor2_0())
|
| 115 |
self.pipe.vae.set_attn_processor(AttnProcessor2_0())
|
| 116 |
if USE_TORCH_COMPILE:
|
|
|
|
| 117 |
self.pipe.unet = torch.compile(self.pipe.unet, mode="reduce-overhead", fullgraph=True)
|
| 118 |
|
| 119 |
if self.realesrgan_x2 is None:
|