Upload folder using huggingface_hub

internals/data/task.py

@@ -145,8 +145,8 @@ class Task:
     def get_high_res_fix(self) -> bool:
         return self.__data.get("high_res_fix", False)
 
-    def get_base_dimension(self) -> int:
-        return self.__data.get("base_dimension", 512)
+    def get_base_dimension(self):
+        return self.__data.get("base_dimension", None)
 
     def get_raw(self) -> dict:
         return self.__data.copy()

internals/util/config.py

@@ -16,6 +16,7 @@ hf_token = base64.b64decode(
     b"aGZfVFZCTHNUam1tT3d6T0h1dlVZWkhEbEZ4WVdOSUdGamVCbA=="
 ).decode()
 hf_cache_dir = "/tmp/hf_hub"
+
 base_dimension = 512  # needed for high res
 
 num_return_sequences = 4  # the number of results to generate
@@ -67,8 +68,10 @@ def get_inpaint_model_path():
 
 
 def get_base_dimension():
-    global base_dimension
-    return base_dimension
+    global global_base_dimension, base_dimension
+    if base_dimension:
+        return base_dimension
+    return model_config.base_dimension  # pyright: ignore
 
 
 def get_is_sdxl():

internals/util/model_loader.py

@@ -15,6 +15,7 @@ class ModelConfig:
     base_model_path: str
     base_inpaint_model_path: str
     is_sdxl: bool = False
+    base_dimension: int = 512
 
 
 def load_model_from_config(path):
@@ -25,10 +26,12 @@ def load_model_from_config(path):
             model_path = config.get("model_path", path)
             inpaint_model_path = config.get("inpaint_model_path", path)
             is_sdxl = config.get("is_sdxl", False)
+            base_dimension = config.get("base_dimension", 512)
 
             m_config.base_model_path = model_path
             m_config.base_inpaint_model_path = inpaint_model_path
             m_config.is_sdxl = is_sdxl
+            m_config.base_dimension = base_dimension
 
             #
             # if config.get("model_type") == "huggingface":