Lykon/dreamshaper-8

app.py

@@ -8,10 +8,10 @@ from PIL import Image
 from diffusers import AutoPipelineForImage2Image
 from diffusers.utils import load_image
 import math
-adapter_id = "latent-consistency/lcm-lora-sdxl"
+adapter_id = "latent-consistency/lcm-lora-sdv1-5"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForImage2Image.from_pretrained("Lykon/dreamshaper-xl-v2-turbo", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
+pipe = AutoPipelineForImage2Image.from_pretrained("Lykon/dreamshaper-8", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
 pipe = pipe.to(device)
 pipe.load_lora_weights(adapter_id)
 pipe.fuse_lora()