Fix varsize to say Do not resize images and use bool
This commit is contained in:
parent
72e962280f
commit
00aa23fbdc
|
@ -294,7 +294,9 @@ class TrainEmbeddingAPI(BaseModel):
|
||||||
)
|
)
|
||||||
training_width: int = Field(title="Training width", description="", default=512)
|
training_width: int = Field(title="Training width", description="", default=512)
|
||||||
training_height: int = Field(title="Training height", 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="")
|
steps: int = Field(title="Max steps", description="")
|
||||||
clip_grad_mode: str = Field(
|
clip_grad_mode: str = Field(
|
||||||
title="Gradient clipping", description="Gradient clip mode"
|
title="Gradient clipping", description="Gradient clip mode"
|
||||||
|
|
Loading…
Reference in New Issue
Block a user