dreamshaper --> sdxl turbo

app/core/schemas.py

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

app/main.py

@@ -24,11 +24,10 @@ 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(
-        'lykon/dreamshaper-xl-v2-turbo', torch_dtype=torch.float16, variant="fp16")
+    ti_pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-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)