Pydantic does't init values in validators I guess

1 year ago · f284983368
parent cd319b0ab1
commit f284983368
1 changed files with 1 additions and 1 deletions
--- a/stabilityai/models.py
+++ b/stabilityai/models.py
@ -192,7 +192,7 @@ class ImageToImageUpscaleRequestBody(BaseModel):
    @validator("width", always=True)
    def mutually_exclusive(cls, v, values):
-        if values["height"] is not None and v:
+        if values.get("height") is not None and v:
            raise ValueError("You can only specify one of width and height.")
        return v