Update app.py

app.py

@@ -13,6 +13,8 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 import torch
+import torch._dynamo
+imoprt diffusers
 from diffusers import AutoencoderKL, StableDiffusionXLPipeline
 
 from diffusers import EulerAncestralDiscreteScheduler
@@ -182,10 +184,12 @@ def load_and_prepare_model():
 
     pipe.vae.set_default_attn_processor()
     pipe.to(device, torch.bfloat16)
+    
     pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", adapter_name="skin")
     #pipe.unet.load_lora_adapter("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", prefix="unet")
     #pipe.text_encoder.load_lora_adapter("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", prefix="text_encoder")
-    
+    pipe.unet = pipe.unet.to(memory_format=torch.contiguous_format)
+
     #pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/epicNewPhoto.safetensors", adapter_name="photo")
     #pipe.set_adapters(["skin", "photo", "fantasy"], adapter_weights=[0.75, 0.25, 0.5])
     #pipe.set_adapters(["skin"], adapter_weights=[0.5])