Update app.py
Browse files
app.py
CHANGED
|
@@ -135,11 +135,9 @@ def load_model():
|
|
| 135 |
pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
|
| 136 |
pipe.to(device=device, dtype=torch.bfloat16)
|
| 137 |
for name, module in pipe.unet.named_modules():
|
| 138 |
-
|
| 139 |
-
|
| 140 |
-
|
| 141 |
upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
|
| 142 |
-
|
| 143 |
return pipe, upscaler_2
|
| 144 |
|
| 145 |
fa_processor = FlashAttentionProcessor()
|
|
|
|
| 135 |
pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
|
| 136 |
pipe.to(device=device, dtype=torch.bfloat16)
|
| 137 |
for name, module in pipe.unet.named_modules():
|
| 138 |
+
if isinstance(module, Attention):
|
| 139 |
+
module.processor = fa_processor
|
|
|
|
| 140 |
upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
|
|
|
|
| 141 |
return pipe, upscaler_2
|
| 142 |
|
| 143 |
fa_processor = FlashAttentionProcessor()
|