HiDream-ai-fast

Paused

blanchon commited on 26 days ago

Commit

fe2f104

1 Parent(s): 4ba46c8

up

Files changed (1) hide show

app-fast.py CHANGED Viewed

@@ -27,13 +27,13 @@ MODEL_CONFIGS = {
 # Supported image sizes
 RESOLUTION_OPTIONS: list[str] = [
-    "1024 x 1024 (Square)",
-    "768 x 1360 (Portrait)",
-    "1360 x 768 (Landscape)",
-    "880 x 1168 (Portrait)",
-    "1168 x 880 (Landscape)",
-    "1248 x 832 (Landscape)",
-    "832 x 1248 (Portrait)",
 ]
 device = torch.device("cuda")

 # Supported image sizes
 RESOLUTION_OPTIONS: list[str] = [
+    "1024 x 1024",
+    "768 x 1360",
+    "1360 x 768",
+    "880 x 1168",
+    "1168 x 880",
+    "1248 x 832",
+    "832 x 1248",
 ]
 device = torch.device("cuda")