roll back and rapply monkey patch

app.py

@@ -8,34 +8,34 @@ from diffusers.pipelines.prx import PRXPipeline
 # monkey patch to add 1024 aspect ratios
 import diffusers.pipelines.prx.pipeline_prx as prx_mod
 
-# CUSTOM_ASPECT_RATIO_512_BIN = {
-#     "0.49": [704, 1440],
-#     "0.52": [736, 1408],
-#     "0.53": [736, 1376],
-#     "0.57": [768, 1344],
-#     "0.59": [768, 1312],
-#     "0.62": [800, 1280],
-#     "0.67": [832, 1248],
-#     "0.68": [832, 1216],
-#     "0.78": [896, 1152],
-#     "0.83": [928, 1120],
-#     "0.94": [992, 1056],
-#     "1.0": [1024, 1024],
-#     "1.06": [1056, 992],
-#     "1.13": [1088, 960],
-#     "1.21": [1120, 928],
-#     "1.29": [1152, 896],
-#     "1.37": [1184, 864],
-#     "1.46": [1216, 832],
-#     "1.5": [1248, 832],
-#     "1.71": [1312, 768],
-#     "1.75": [1344, 768],
-#     "1.87": [1376, 736],
-#     "1.91": [1408, 736],
-#     "2.05": [1440, 704],
-# }
-
-# prx_mod.ASPECT_RATIO_512_BIN = CUSTOM_ASPECT_RATIO_512_BIN
+CUSTOM_ASPECT_RATIO_512_BIN = {
+    "0.49": [704, 1440],
+    "0.52": [736, 1408],
+    "0.53": [736, 1376],
+    "0.57": [768, 1344],
+    "0.59": [768, 1312],
+    "0.62": [800, 1280],
+    "0.67": [832, 1248],
+    "0.68": [832, 1216],
+    "0.78": [896, 1152],
+    "0.83": [928, 1120],
+    "0.94": [992, 1056],
+    "1.0": [1024, 1024],
+    "1.06": [1056, 992],
+    "1.13": [1088, 960],
+    "1.21": [1120, 928],
+    "1.29": [1152, 896],
+    "1.37": [1184, 864],
+    "1.46": [1216, 832],
+    "1.5": [1248, 832],
+    "1.71": [1312, 768],
+    "1.75": [1344, 768],
+    "1.87": [1376, 736],
+    "1.91": [1408, 736],
+    "2.05": [1440, 704],
+}
+
+prx_mod.ASPECT_RATIO_512_BIN = CUSTOM_ASPECT_RATIO_512_BIN
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"