Upload folder using huggingface_hub

README.md

@@ -18,6 +18,7 @@ models:
   - black-forest-labs/FLUX.2-klein-base-4b-fp8
   - black-forest-labs/FLUX.2-klein-base-9b-fp8
   - bluepen5805/4nima_pencil-XL
+  - bluepen5805/anima-models
   - bluepen5805/anima_pencil-XL
   - bluepen5805/blue_pencil-XL
   - bluepen5805/illustrious_pencil-XL

comfy_integration/setup.py

@@ -6,11 +6,13 @@ from core.settings import *
 
 def move_and_overwrite(src, dst):
     if os.path.isdir(src):
-        if os.path.exists(dst):
-            shutil.rmtree(dst)
-        shutil.move(src, dst)
+        os.makedirs(dst, exist_ok=True)
+        for item in os.listdir(src):
+            s = os.path.join(src, item)
+            d = os.path.join(dst, item)
+            move_and_overwrite(s, d)
     elif os.path.isfile(src):
-        if os.path.exists(dst):
+        if os.path.exists(dst) or os.path.islink(dst):
             os.remove(dst)
         shutil.move(src, dst)
 
@@ -89,6 +91,29 @@ def initialize_comfyui():
 
     print("✅ ComfyUI initialized with default attention mechanism.")
 
+    def create_model_symlink(volume_path, target_path):
+        if os.path.exists(volume_path):
+            os.makedirs(os.path.dirname(target_path), exist_ok=True)
+            
+            if os.path.exists(target_path) or os.path.islink(target_path):
+                if os.path.islink(target_path):
+                    os.remove(target_path)
+                elif os.path.isdir(target_path):
+                    shutil.rmtree(target_path)
+                else:
+                    os.remove(target_path)
+                    
+            try:
+                os.symlink(volume_path, target_path)
+                print(f"✅ Created symlink from {volume_path} to {target_path}")
+            except Exception as e:
+                print(f"❌ Failed to create symlink: {e}")
+        else:
+            print(f"⚠️ Warning: Volume mount path '{volume_path}' not found.")
+
+    create_model_symlink("/home/user/volume/text_encoders", os.path.join(APP_DIR, "models", "text_encoders"))
+    create_model_symlink("/home/user/volume/diffusion_models", os.path.join(APP_DIR, "models", "diffusion_models"))
+
     for dir_path in CATEGORY_TO_DIR_MAP.values():
         os.makedirs(os.path.join(APP_DIR, dir_path), exist_ok=True)
     

yaml/file_list.yaml

@@ -403,16 +403,6 @@ file:
       repo_id: "comfyanonymous/ControlNet-v1-1_fp16_safetensors"
       repository_file_path: "control_v11u_sd15_tile_fp16.safetensors"
   diffusion_models: 
-    # FLUX.2-klein-9B
-    - filename: "flux-2-klein-9b-fp8.safetensors"
-      source: "hf"
-      repo_id: "black-forest-labs/FLUX.2-klein-9b-fp8"
-      repository_file_path: "flux-2-klein-9b-fp8.safetensors"
-    # FLUX.2-klein-base-9B
-    - filename: "flux-2-klein-base-9b-fp8.safetensors"
-      source: "hf"
-      repo_id: "black-forest-labs/FLUX.2-klein-base-9b-fp8"
-      repository_file_path: "flux-2-klein-base-9b-fp8.safetensors"
     # Anima
     - filename: "waiANIMA_v10.safetensors"
       source: "hf"
@@ -426,6 +416,10 @@ file:
       source: "hf"
       repo_id: "duongve/AnimaYume"
       repository_file_path: "split_files/diffusion_models/AnimaYume_tuned_v04.safetensors"
+    - filename: "anima_pencil-v1.0.safetensors"
+      source: "hf"
+      repo_id: "bluepen5805/anima-models"
+      repository_file_path: "anima_pencil-v1.0.safetensors"
     # NewBie-Image
     - filename: "NewBie-Image-Exp0.1-bf16.safetensors"
       source: "hf"
@@ -536,10 +530,6 @@ file:
       source: "hf"
       repo_id: "Comfy-Org/HunyuanImage_2.1_ComfyUI"
       repository_file_path: "split_files/diffusion_models/hunyuanimage2.1_distilled_fp8_e4m3fn.safetensors"
-    - filename: "hunyuanimage2.1_refiner_fp8_e4m3fn.safetensors"
-      source: "hf"
-      repo_id: "Comfy-Org/HunyuanImage_2.1_ComfyUI"
-      repository_file_path: "split_files/diffusion_models/hunyuanimage2.1_refiner_fp8_e4m3fn.safetensors" 
     # Chroma1-Radiance
     - filename: "Chroma1-Radiance-x0-fp8mixed_fullmm-20260104.safetensors"
       source: "hf"

yaml/model_list.yaml

@@ -97,6 +97,11 @@ Checkpoint:
           unet: "AnimaYume_tuned_v04.safetensors"
           vae: "qwen_image_vae.safetensors"
           clip: "qwen_3_06b_base.safetensors"
+      - display_name: "bluepen5805/Anima-pencil-v1.0"
+        components:
+          unet: "anima_pencil-v1.0.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"