Upload folder using huggingface_hub

requirements.txt

@@ -1,6 +1,6 @@
-comfyui-frontend-package==1.45.19
+comfyui-frontend-package==1.45.20
 comfyui-workflow-templates==0.10.7
-comfyui-embedded-docs==0.5.5
+comfyui-embedded-docs==0.5.6
 torch
 torchsde
 torchvision
@@ -22,7 +22,7 @@ alembic
 SQLAlchemy>=2.0.0
 filelock
 av>=16.0.0
-comfy-kitchen==0.2.14
+comfy-kitchen==0.2.15
 comfy-aimdo==0.4.10
 requests
 simpleeval>=1.0.0

utils/app_utils.py

@@ -129,12 +129,20 @@ def get_civitai_file_info(version_id: str) -> dict | None:
         response.raise_for_status()
         data = response.json()
         
+        model_type = data.get('model', {}).get('type')
+        
+        result_file = None
         for file_data in data.get('files', []):
             if file_data.get('type') == 'Model' and file_data['name'].endswith(('.safetensors', '.pt', '.bin')):
-                return file_data
+                result_file = file_data.copy()
+                break
         
-        if data.get('files'):
-            return data['files'][0]
+        if not result_file and data.get('files'):
+            result_file = data['files'][0].copy()
+            
+        if result_file:
+            result_file['model_type'] = model_type
+            return result_file
     except Exception:
         return None
 
@@ -173,9 +181,15 @@ def get_lora_path(source: str, id_or_url: str, civitai_key: str, progress) -> tu
             version_id = sanitize_id(id_or_url)
             if not version_id:
                 return None, "Invalid Civitai ID provided. Must be numeric."
+            
+            file_info = get_civitai_file_info(version_id)
+            if file_info:
+                model_type = file_info.get('model_type')
+                if not model_type or model_type.lower() not in ['lora', 'lycoris', 'dora']:
+                    return None, f"Invalid Civitai model type '{model_type}' for LoRA. Allowed types: LoRA, LyCORIS, DoRA."
+            
             filename = sanitize_filename(f"civitai_{version_id}.safetensors")
             local_path = os.path.join(LORA_DIR, filename)
-            file_info = get_civitai_file_info(version_id)
             api_key_to_use = civitai_key
             source_name = f"Civitai ID {version_id}"
         elif source == "Hugging Face":
@@ -223,13 +237,18 @@ def get_embedding_path(source: str, id_or_url: str, civitai_key: str, progress)
 
     try:
         if source == "Civitai":
-            file_ext = ".safetensors"
             version_id = sanitize_id(id_or_url)
             if not version_id:
                 return None, "Invalid Civitai ID. Must be numeric."
             
             file_info = get_civitai_file_info(version_id)
-            if file_info and file_info['name'].lower().endswith(('.pt', '.bin')):
+            if file_info:
+                model_type = file_info.get('model_type')
+                if not model_type or model_type.lower() not in ['embedding', 'textualinversion']:
+                    return None, f"Invalid Civitai model type '{model_type}' for Embedding. Allowed types: Embedding."
+            
+            file_ext = ".safetensors"
+            if file_info and file_info.get('name') and file_info['name'].lower().endswith(('.pt', '.bin')):
                 file_ext = os.path.splitext(file_info['name'])[1]
 
             filename = sanitize_filename(f"civitai_{version_id}{file_ext}")
@@ -280,13 +299,18 @@ def get_vae_path(source: str, id_or_url: str, civitai_key: str, progress) -> tup
 
     try:
         if source == "Civitai":
-            file_ext = ".safetensors"
             version_id = sanitize_id(id_or_url)
             if not version_id:
                 return None, "Invalid Civitai ID. Must be numeric."
             
             file_info = get_civitai_file_info(version_id)
-            if file_info and file_info['name'].lower().endswith(('.pt', '.bin')):
+            if file_info:
+                model_type = file_info.get('model_type')
+                if not model_type or model_type.lower() != 'vae':
+                    return None, f"Invalid Civitai model type '{model_type}' for VAE. Allowed types: VAE."
+            
+            file_ext = ".safetensors"
+            if file_info and file_info.get('name') and file_info['name'].lower().endswith(('.pt', '.bin')):
                 file_ext = os.path.splitext(file_info['name'])[1]
 
             filename = sanitize_filename(f"civitai_{version_id}{file_ext}")

yaml/file_list.yaml

@@ -475,18 +475,26 @@ file:
       repo_id: "nvidia/Cosmos-Predict2-14B-Text2Image"
       repository_file_path: "model.pt"
     # Anima
-    - filename: "waiANIMA_v10.safetensors"
+    - filename: "waiANIMA_v10Base10.safetensors"
       source: "hf"
       repo_id: "diffusionmodels1254ani/waiANIMA"
-      repository_file_path: "waiANIMA_v10.safetensors"
+      repository_file_path: "waiANIMA_v10Base10.safetensors"
     - filename: "anima-base-v1.0.safetensors"
       source: "hf"
       repo_id: "circlestone-labs/Anima"
       repository_file_path: "split_files/diffusion_models/anima-base-v1.0.safetensors"
-    - filename: "AnimaYume_v10_base.safetensors"
+    - filename: "AnimaYume_v10_final_base.safetensors"
       source: "hf"
       repo_id: "duongve/AnimaYume"
-      repository_file_path: "split_files/diffusion_models/AnimaYume_v10_base.safetensors"
+      repository_file_path: "split_files/diffusion_models/AnimaYume_v10_final_base.safetensors"
+    - filename: "hassakuAnima_v1Style.safetensors"
+      source: "hf"
+      repo_id: "diffusionmodels1254ani/hassakuAnima"
+      repository_file_path: "hassakuAnima_v1Style.safetensors"
+    - filename: "kirazuriAnima_v30AnimaBase1.safetensors"
+      source: "hf"
+      repo_id: "diffusionmodels1254ani/kirazuriAnima_v30AnimaBase1"
+      repository_file_path: "kirazuriAnima_v30AnimaBase1.safetensors"
     - filename: "anima_pencil-v2.1.0.safetensors"
       source: "hf"
       repo_id: "bluepen5805/anima-models"

yaml/model_list.yaml

@@ -156,12 +156,12 @@ Checkpoint:
     models:
       - display_name: "WAI0731/waiANIMA-v1.0"
         components:
-          unet: "waiANIMA_v10.safetensors"
+          unet: "waiANIMA_v10Base10.safetensors"
           vae: "qwen_image_vae.safetensors"
           clip: "qwen_3_06b_base.safetensors"
       - display_name: "duongve/AnimaYume-v1.0"
         components:
-          unet: "AnimaYume_v10_base.safetensors"
+          unet: "AnimaYume_v10_final_base.safetensors"
           vae: "qwen_image_vae.safetensors"
           clip: "qwen_3_06b_base.safetensors"
       - display_name: "bluepen5805/Anima-pencil-v2.1"
@@ -169,6 +169,16 @@ Checkpoint:
           unet: "anima_pencil-v2.1.0.safetensors"
           vae: "qwen_image_vae.safetensors"
           clip: "qwen_3_06b_base.safetensors"
+      - display_name: "Ikena/Hassaku-Anima-v1-Style"
+        components:
+          unet: "hassakuAnima_v1Style.safetensors"
+          vae: "qwen_image_vae.safetensors"
+          clip: "qwen_3_06b_base.safetensors"
+      - display_name: "motimalu/Kirazuri (Anima)-v3.0"
+        components:
+          unet: "hassakuAnima_v1Style.safetensors"
+          vae: "qwen_image_vae.safetensors"
+          clip: "qwen_3_06b_base.safetensors"
       - display_name: "circlestone-labs/Anima-base-v1.0"
         components:
           unet: "anima-base-v1.0.safetensors"