ADD Option: OFFLINE fetch modelscope model (#172)

* ADD option: fetch modelscope model OFFLINE

* format

* Apply suggestion from @gemini-code-assist[bot]

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

---------

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: qzzz95

diffsynth_engine/utils/download.py

@@ -12,7 +12,7 @@
 from modelscope.hub.api import HubApi
 from diffsynth_engine.utils import logging
 from diffsynth_engine.utils.lock import HeartbeatFileLock
-from diffsynth_engine.utils.env import DIFFSYNTH_FILELOCK_DIR, DIFFSYNTH_CACHE
+from diffsynth_engine.utils.env import DIFFSYNTH_FILELOCK_DIR, DIFFSYNTH_CACHE, MS_HUB_OFFLINE
 from diffsynth_engine.utils.constants import MB
 
 logger = logging.get_logger(__name__)
@@ -81,7 +81,13 @@ def fetch_modelscope_model(
         api.login(access_token)
     with HeartbeatFileLock(lock_file_path):
         directory = os.path.join(DIFFSYNTH_CACHE, "modelscope", model_id, revision if revision else "__version")
-        dirpath = snapshot_download(model_id, revision=revision, local_dir=directory, allow_patterns=path)
+        dirpath = snapshot_download(
+            model_id,
+            revision=revision,
+            local_dir=directory,
+            allow_patterns=path,
+            local_files_only=MS_HUB_OFFLINE
+        )
 
     if isinstance(path, str):
         path = glob.glob(os.path.join(dirpath, path))

diffsynth_engine/utils/env.py

@@ -8,3 +8,5 @@
 DIFFSYNTH_FILELOCK_DIR = os.environ.get(
     "DIFFSYNTH_FILELOCK_DIR", os.path.join(HOME, ".cache", "diffsynth", "filelocks")
 )
+
+MS_HUB_OFFLINE = os.getenv("MS_HUB_OFFLINE", "0").lower() in ("1", "true", "yes")