update some codes related to win32

modules_forge/patch_basic.py

@@ -6,6 +6,7 @@ import warnings
 import gradio.networking
 import safetensors.torch
 
+from pathlib import Path
 from tqdm import tqdm
 
 
@@ -64,6 +65,12 @@ def always_show_tqdm(*args, **kwargs):
     return tqdm(*args, **kwargs)
 
 
+def long_path_prefix(path: Path) -> Path:
+    if os.name == 'nt' and not str(path).startswith("\\\\?\\") and not path.exists():
+        return Path("\\\\?\\" + str(path))
+    return path
+
+
 def patch_all_basics():
     import logging
     from huggingface_hub import file_download
@@ -71,6 +78,15 @@ def patch_all_basics():
     from transformers.dynamic_module_utils import logger
     logger.setLevel(logging.ERROR)
 
+    from huggingface_hub.file_download import _download_to_tmp_and_move as original_download_to_tmp_and_move
+
+    def patched_download_to_tmp_and_move(incomplete_path, destination_path, url_to_download, proxies, headers, expected_size, filename, force_download):
+        incomplete_path = long_path_prefix(incomplete_path)
+        destination_path = long_path_prefix(destination_path)
+        return original_download_to_tmp_and_move(incomplete_path, destination_path, url_to_download, proxies, headers, expected_size, filename, force_download)
+
+    file_download._download_to_tmp_and_move = patched_download_to_tmp_and_move
+
     gradio.networking.url_ok = gradio_url_ok_fix
     build_loaded(safetensors.torch, 'load_file')
     build_loaded(torch, 'load')