Upload folder using huggingface_hub

utils/app_utils.py

@@ -105,6 +105,14 @@ def sanitize_prompt(prompt: str) -> str:
 def sanitize_id(input_id: str) -> str:
     if not isinstance(input_id, str):
         return ""
+    input_id = input_id.strip()
+    if "civitai" in input_id.lower():
+        version_match = re.search(r'modelVersionId=(\d+)', input_id)
+        if version_match:
+            return version_match.group(1)
+        model_match = re.search(r'/models/(\d+)', input_id)
+        if model_match:
+            return model_match.group(1)
     return re.sub(r'[^0-9]', '', input_id)
 
 def sanitize_url(url: str) -> str:
@@ -356,7 +364,6 @@ def get_vae_path(source: str, id_or_url: str, civitai_key: str, progress) -> tup
         except Exception as e:
             return None, f"Hugging Face download failed: {e}"
 
-
 def _ensure_model_downloaded(display_name: str, progress=gr.Progress()):
     if display_name not in ALL_MODEL_MAP:
         raise ValueError(f"Model '{display_name}' not found in configuration.")
@@ -549,7 +556,6 @@ def ensure_ipadapter_models_downloaded(preset_name: str, progress):
         except Exception as e:
             print(f"❌ Error ensuring download for IPAdapter asset '{filename}': {e}")
 
-
 def ensure_sd3_ipadapter_models_downloaded(progress):
     _PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
     yaml_path = os.path.join(_PROJECT_ROOT, 'yaml', 'ipadapter_sd3_models.yaml')
@@ -564,8 +570,6 @@ def ensure_sd3_ipadapter_models_downloaded(progress):
     except Exception as e:
         print(f"Warning: Failed to load or download sd3 ipadapter models: {e}")
 
-
-
 def get_model_generation_defaults(model_display_name: str, model_type: str, defaults_config: dict):
     final_defaults = {
         'steps': 25, 'cfg': 7.0, 'sampler_name': 'euler', 'scheduler': 'simple',

yaml/constants.yaml

@@ -247,7 +247,7 @@ MULTIPLIERS_MAP:
   omnigen2: 1
   lumina: 1
   hidream-o1: 32
-  hidream-i1:
+  hidream-i1: 1
   flux1: 1
   sd35: 1
   sdxl: 1