Lykon/dreamshaper-xl-lightning

app.py

@@ -13,7 +13,7 @@ from DeepCache import DeepCacheSDHelper
 adapter_id = "latent-consistency/lcm-lora-sdv1-5"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = OVStableDiffusionImg2ImgPipeline.from_pretrained("Lykon/dreamshaper-xl-v2-turbo", torch_dtype=torch.float16, export=True) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
+pipe = OVStableDiffusionImg2ImgPipeline.from_pretrained("Lykon/dreamshaper-xl-lightning", torch_dtype=torch.float16, export=True) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
 pipe = pipe.to(device)
 # helper = DeepCacheSDHelper(pipe=pipe)
 # helper.set_params(