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:
|