forced height wnd width

f06ba50f · novelailab · deff8bc2 · f06ba50f
Commit f06ba50f authored Aug 04, 2022 by novelailab
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

main.py main.py +2 -2

No files found.
--- a/main.py
+++ b/main.py
@@ -56,8 +56,8 @@ class GenerationRequest(BaseModel):
    plms: bool = True
    fixed_code: bool = False
    ddim_eta: float = 0.0
-    height: int
-    width: int
+    height: int = 512
+    width: int = 512
    latent_channels: int = 4
    downsampling_factor: int = 8
    scale: float = 7.0