SDXL LORA support

app.py

@@ -129,6 +129,13 @@ elif model_type == "LORA":
             use_safetensors=True,
             variant="fp16",
         )
+    elif base_model == "SDXL 1.0":
+        pipe = AutoPipelineForImage2Image.from_pretrained(
+            "stabilityai/stable-diffusion-xl-refiner-1.0",
+            torch_dtype=torch.float16,
+            use_safetensors=True,
+            variant="fp16",
+        )
     else:
         raise ValueError(f"Unsupported base model: {base_model}")
 
@@ -186,7 +193,7 @@ css = """
 
 with gr.Blocks(css=css) as demo:
     with gr.Column():
-        gr.Markdown("# Image-to-Image")
+        gr.Markdown("# Image-to-Image with Civitai Models")
         gr.Markdown(f"## Model: [{model_name}]({model_url})")
 
         with gr.Row():