Update main/load.py

main/load.py

@@ -24,7 +24,7 @@ config = {
         "embed_dim": 4,
         "double_z": True,
         "z_channels": 4,
-        "resolution": 256,
+        "resolution": 512,
         "in_channels": 4,
         "out_ch": 4,
         "ch": 128,
@@ -34,7 +34,7 @@ config = {
         "dropout": 0.0
     },
     "cond_stage_config": {
-        "image_size": 64,
+        "image_size": 128,
         "in_channels": 8,
         "model_channels": 160,
         "out_channels": 4,