Update app.py
Browse files
app.py
CHANGED
|
@@ -163,7 +163,6 @@ def load_and_prepare_model():
|
|
| 163 |
|
| 164 |
#pipe.vae.force_upcast=True
|
| 165 |
# pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/Fantasy_World_XL.safetensors", adapter_name="fantasy")
|
| 166 |
-
pipe.vae.set_default_attn_processor() # Set attention processor first
|
| 167 |
pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", adapter_name="skin", low_cpu_mem_usage=False,token=HF_TOKEN)
|
| 168 |
|
| 169 |
#pipe.unet = pipe.unet.to(memory_format=torch.contiguous_format)
|
|
@@ -181,7 +180,6 @@ def load_and_prepare_model():
|
|
| 181 |
pipe.vae.set_default_attn_processor()
|
| 182 |
#pipe.to(device)
|
| 183 |
#pipe.to(torch.bfloat16)
|
| 184 |
-
|
| 185 |
print(f'Pipeline: ')
|
| 186 |
#print(f'_optional_components: {pipe._optional_components}')
|
| 187 |
#print(f'watermark: {pipe.watermark}')
|
|
|
|
| 163 |
|
| 164 |
#pipe.vae.force_upcast=True
|
| 165 |
# pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/Fantasy_World_XL.safetensors", adapter_name="fantasy")
|
|
|
|
| 166 |
pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", adapter_name="skin", low_cpu_mem_usage=False,token=HF_TOKEN)
|
| 167 |
|
| 168 |
#pipe.unet = pipe.unet.to(memory_format=torch.contiguous_format)
|
|
|
|
| 180 |
pipe.vae.set_default_attn_processor()
|
| 181 |
#pipe.to(device)
|
| 182 |
#pipe.to(torch.bfloat16)
|
|
|
|
| 183 |
print(f'Pipeline: ')
|
| 184 |
#print(f'_optional_components: {pipe._optional_components}')
|
| 185 |
#print(f'watermark: {pipe.watermark}')
|