Lykon/dreamshaper-xl-v2-turbo

app.py

@@ -8,12 +8,19 @@ from PIL import Image
 from diffusers import AutoPipelineForImage2Image
 from diffusers.utils import load_image
 import math
+from DeepCache import DeepCacheSDHelper
+
 adapter_id = "latent-consistency/lcm-lora-sdv1-5"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForImage2Image.from_pretrained("Lykon/dreamshaper-8", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
+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 = pipe.to(device)
-
+helper = DeepCacheSDHelper(pipe=pipe)
+helper.set_params(
+    cache_interval=3,
+    cache_branch_id=0,
+)
+helper.enable()
 def resize(value,img):
     img = Image.open(img)
     img = img.resize((value,value))