From 00aa23fbdcc4a64b34c0cc3a0f89d6381023ffc8 Mon Sep 17 00:00:00 2001 From: "Dave Lage (rockerBOO)" Date: Thu, 26 Jan 2023 03:59:12 -0500 Subject: [PATCH] Fix varsize to say Do not resize images and use bool --- modules/api/models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modules/api/models.py b/modules/api/models.py index ebf0d3c8..560b5918 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -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"