eienmojiki commited on
Commit
c5a8f52
·
verified ·
1 Parent(s): 6aa4df8

Update models/text_to_image.py

Browse files
Files changed (1) hide show
  1. models/text_to_image.py +1 -1
models/text_to_image.py CHANGED
@@ -4,7 +4,7 @@ from typing import List, Optional
4
 
5
  class TextToImageRequest(BaseModel):
6
  prompt: str = Field(..., description="The prompt to generate an image from.")
7
- negative_prompt: Optional[List[str]] = Field(None, description="One or several prompts to guide what NOT to include in image generation.")
8
  height: Optional[int] = Field(None, description="The height in pixels of the image to generate.", ge=64, le=2048)
9
  width: Optional[int] = Field(None, description="The width in pixels of the image to generate.", ge=64, le=2048)
10
  num_inference_steps: Optional[int] = Field(None, description="The number of denoising steps. More denoising steps usually lead to a higher quality image at the expense of slower inference.", ge=1, le=500)
 
4
 
5
  class TextToImageRequest(BaseModel):
6
  prompt: str = Field(..., description="The prompt to generate an image from.")
7
+ negative_prompt: Optional[str] = Field(None, description="One or several prompts to guide what NOT to include in image generation.")
8
  height: Optional[int] = Field(None, description="The height in pixels of the image to generate.", ge=64, le=2048)
9
  width: Optional[int] = Field(None, description="The width in pixels of the image to generate.", ge=64, le=2048)
10
  num_inference_steps: Optional[int] = Field(None, description="The number of denoising steps. More denoising steps usually lead to a higher quality image at the expense of slower inference.", ge=1, le=500)