Use base model for SDXL

app.py

@@ -149,8 +149,10 @@ elif model_type == "LORA":
             variant="fp16",
         )
     elif base_model == "SDXL 1.0":
+        # Use AutoPipelineForImage2Image with the base model
+        # since LORA are trained on base
         pipe = AutoPipelineForImage2Image.from_pretrained(
-            "stabilityai/stable-diffusion-xl-refiner-1.0",
+            "stabilityai/stable-diffusion-xl-base-1.0",
             torch_dtype=torch.float16,
             use_safetensors=True,
             variant="fp16",