sdxl turbo -> dreamshaper

app/core/schemas.py

@@ -55,8 +55,8 @@ class Generate(BaseModel):
 
 class TextImage(Generate):
     prompt: str
-    num_inference_steps: int = 1
-    guidance_scale: float = 0.0
+    num_inference_steps: int = 4
+    guidance_scale: float = 2.0
 
 
 class ImageImage(Generate):

app/main.py

@@ -24,10 +24,11 @@ async def lifespan(app: FastAPI):
     image_encoder_path = "sdxl_models/image_encoder"
     ip_ckpt = "sdxl_models/ip-adapter_sdxl.bin"
 
-    ti_pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
+    ti_pipe = AutoPipelineForText2Image.from_pretrained(
+        'lykon/dreamshaper-xl-v2-turbo', torch_dtype=torch.float16, variant="fp16")
     ti_pipe.to(DEVICE)
-    # ti_pipe.scheduler = DPMSolverMultistepScheduler.from_config(
-    #     ti_pipe.scheduler.config)
+    ti_pipe.scheduler = DPMSolverMultistepScheduler.from_config(
+        ti_pipe.scheduler.config)
 
     ii_pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
         "timbrooks/instruct-pix2pix", torch_dtype=torch.float16, safety_checker=None)