Fix varsize to say Do not resize images and use bool

This commit is contained in:
Dave Lage (rockerBOO) 2023-01-26 03:59:12 -05:00
parent 72e962280f
commit 00aa23fbdc
No known key found for this signature in database
GPG Key ID: 0D4EAF00DCABC97B

View File

@ -294,7 +294,9 @@ class TrainEmbeddingAPI(BaseModel):
)
training_width: int = Field(title="Training width", description="", default=512)
training_height: int = Field(title="Training height", description="", default=512)
varsize: str = Field(title="", description="")
varsize: bool = Field(
title="Do not resize images", description="Do not resize images", default=False
)
steps: int = Field(title="Max steps", description="")
clip_grad_mode: str = Field(
title="Gradient clipping", description="Gradient clip mode"