ford442 commited on
Commit
873e499
·
verified ·
1 Parent(s): 5485da0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -173,7 +173,8 @@ def load_and_prepare_model():
173
  #pipe.scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1)
174
  #pipe.scheduler=EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
175
 
176
- pipe.to(device=device, dtype=torch.bfloat16)
 
177
 
178
  #pipe.unet.set_default_attn_processor()
179
  pipe.vae.set_default_attn_processor()
@@ -187,7 +188,6 @@ def load_and_prepare_model():
187
  #print(f'UNET: {pipe.unet}')
188
  pipe.watermark=None
189
  pipe.safety_checker=None
190
-
191
 
192
  # pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/Fantasy_World_XL.safetensors", adapter_name="fantasy")
193
  pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", adapter_name="skin")
 
173
  #pipe.scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1)
174
  #pipe.scheduler=EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
175
 
176
+ pipe.to(device)
177
+ pipe.to(torch.bfloat16)
178
 
179
  #pipe.unet.set_default_attn_processor()
180
  pipe.vae.set_default_attn_processor()
 
188
  #print(f'UNET: {pipe.unet}')
189
  pipe.watermark=None
190
  pipe.safety_checker=None
 
191
 
192
  # pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/Fantasy_World_XL.safetensors", adapter_name="fantasy")
193
  pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", adapter_name="skin")