1inkusFace commited on
Commit
fe7a814
·
verified ·
1 Parent(s): e95c348

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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
- if isinstance(module, Attention):
139
- module.processor = fa_processor
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()