change base model

app_text2img.py

@@ -20,7 +20,7 @@ from controlnet_aux import (
     CannyDetector,
 )
 
-BASE_MODEL = "SG161222/RealVisXL_V5.0_Lightning"
+BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 DEFAULT_EDIT_PROMPT = "a woman, blue hair, high detailed"
@@ -88,12 +88,12 @@ def image_to_image(
         generator=generator,
         prompt=edit_prompt,
         negative_prompt=DEFAULT_NEGATIVE_PROMPT,
-        image=cond_image,
+        # image=cond_image,
         height=generate_size,
         width=generate_size,
         guidance_scale=guidance_scale,
         num_inference_steps=num_steps,
-        # adapter_image=cond_image,
+        adapter_image=cond_image,
         adapter_conditioning_scale=cond_scale,
     ).images[0]