From 1313f4dd63d3e46d7dac532083f4886118641262 Mon Sep 17 00:00:00 2001
From: Artiprocher <wangye87v5@hotmail.com>
Date: Tue, 18 Nov 2025 10:29:07 +0800
Subject: [PATCH] support downloading resource

---
 diffsynth/utils/__init__.py | 26 +++++++++++++++++++-------
 setup.py                    |  2 +-
 2 files changed, 20 insertions(+), 8 deletions(-)

diff --git a/diffsynth/utils/__init__.py b/diffsynth/utils/__init__.py
index ec3c727..133f899 100644
--- a/diffsynth/utils/__init__.py
+++ b/diffsynth/utils/__init__.py
@@ -4,6 +4,7 @@ from PIL import Image
 from einops import repeat, reduce
 from typing import Optional, Union
 from dataclasses import dataclass
+from huggingface_hub import snapshot_download as hf_snapshot_download
 from modelscope import snapshot_download
 import numpy as np
 from PIL import Image
@@ -196,13 +197,24 @@ class ModelConfig:
                 self.local_model_path = "./models"
             if not skip_download:
                 downloaded_files = glob.glob(self.origin_file_pattern, root_dir=os.path.join(self.local_model_path, self.model_id))
-                snapshot_download(
-                    self.model_id,
-                    local_dir=os.path.join(self.local_model_path, self.model_id),
-                    allow_file_pattern=allow_file_pattern,
-                    ignore_file_pattern=downloaded_files,
-                    local_files_only=False
-                )
+                if self.download_resource.lower() == "modelscope":
+                    snapshot_download(
+                        self.model_id,
+                        local_dir=os.path.join(self.local_model_path, self.model_id),
+                        allow_file_pattern=allow_file_pattern,
+                        ignore_file_pattern=downloaded_files,
+                        local_files_only=False
+                    )
+                elif self.download_resource.lower() == "huggingface":
+                    hf_snapshot_download(
+                        self.model_id,
+                        local_dir=os.path.join(self.local_model_path, self.model_id),
+                        allow_patterns=allow_file_pattern,
+                        ignore_patterns=downloaded_files,
+                        local_files_only=False
+                    )
+                else:
+                    raise ValueError("`download_resource` should be `modelscope` or `huggingface`.")
             
             # Let rank 1, 2, ... wait for rank 0
             if use_usp:
diff --git a/setup.py b/setup.py
index b495a8f..e72b901 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@ else:
 
 setup(
     name="diffsynth",
-    version="1.1.8",
+    version="1.1.9",
     description="Enjoy the magic of Diffusion models!",
     author="Artiprocher",
     packages=find_packages(),