+

models/text_to_image.py

@@ -11,5 +11,5 @@ class TextToImageRequest(BaseModel):
     guidance_scale: Optional[float] = Field(None, description="A higher guidance scale value encourages the model to generate images closely linked to the text prompt, but values too high may cause saturation and other artifacts.")
     model: Optional[str] = Field(None, description="The model to use for inference. Can be a model ID hosted on the Hugging Face Hub or a URL to a deployed Inference Endpoint. If not provided, the default recommended text-to-image model will be used.")
     scheduler: Optional[str] = Field(None, description="Override the scheduler with a compatible one.")
-    target_size: Optional[TextToImageTargetSize] = Field(None, description="The size in pixel of the output image")
+    #target_size: Optional[TextToImageTargetSize] = Field(None, description="The size in pixel of the output image")
     seed: Optional[int] = Field(None, description="Seed for the random number generator.")

routes/textToImage.py

@@ -16,7 +16,7 @@ async def text_to_image(body: TextToImageRequest):
         num_inference_steps=body.num_inference_steps,
         guidance_scale=body.guidance_scale,
         scheduler=body.scheduler,
-        target_size=body.target_size,
+       # target_size=body.target_size,
         seed=body.seed
     )
     img_byte_arr = io.BytesIO()