fixes

src/fastserve/models/__main__.py

@@ -46,6 +46,7 @@
 
 if args.model == "ssd-1b":
     app = ServeSSD1B(device=device, timeout=args.timeout, batch_size=args.batch_size)
+
 elif args.model == "sdxl-turbo":
     app = ServeSDXLTurbo(
         device=device, timeout=args.timeout, batch_size=args.batch_size

src/fastserve/models/sdxl_turbo.py

@@ -25,12 +25,11 @@ def __init__(
                 "It is recommended to use inference_steps=1 for SDXL Turbo model."
             )
         self.num_inference_steps = num_inference_steps
-        self.input_schema = PromptRequest
         self.pipe = AutoPipelineForText2Image.from_pretrained(
             "stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16"
         )
         self.pipe.to(device)
-        super().__init__(batch_size, timeout)
+        super().__init__(batch_size, timeout, input_schema=PromptRequest)
 
     def handle(self, batch: List[PromptRequest]) -> List[StreamingResponse]:
         prompts = [b.prompt for b in batch]