Skip to content

Commit 261a448

Browse files
Correct hf hub download (#1767)
* allow model download when no internet * up * make style
1 parent f106ab4 commit 261a448

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/diffusers/modeling_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
from requests import HTTPError
2727

2828
from . import __version__
29+
from .hub_utils import HF_HUB_OFFLINE
2930
from .utils import (
3031
CONFIG_NAME,
3132
DIFFUSERS_CACHE,
@@ -376,7 +377,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
376377
resume_download = kwargs.pop("resume_download", False)
377378
proxies = kwargs.pop("proxies", None)
378379
output_loading_info = kwargs.pop("output_loading_info", False)
379-
local_files_only = kwargs.pop("local_files_only", False)
380+
local_files_only = kwargs.pop("local_files_only", HF_HUB_OFFLINE)
380381
use_auth_token = kwargs.pop("use_auth_token", None)
381382
revision = kwargs.pop("revision", None)
382383
torch_dtype = kwargs.pop("torch_dtype", None)

src/diffusers/pipeline_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333

3434
from .configuration_utils import ConfigMixin
3535
from .dynamic_modules_utils import get_class_from_dynamic_module
36-
from .hub_utils import http_user_agent
36+
from .hub_utils import HF_HUB_OFFLINE, http_user_agent
3737
from .modeling_utils import _LOW_CPU_MEM_USAGE_DEFAULT
3838
from .schedulers.scheduling_utils import SCHEDULER_CONFIG_NAME
3939
from .utils import (
@@ -441,7 +441,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
441441
resume_download = kwargs.pop("resume_download", False)
442442
force_download = kwargs.pop("force_download", False)
443443
proxies = kwargs.pop("proxies", None)
444-
local_files_only = kwargs.pop("local_files_only", False)
444+
local_files_only = kwargs.pop("local_files_only", HF_HUB_OFFLINE)
445445
use_auth_token = kwargs.pop("use_auth_token", None)
446446
revision = kwargs.pop("revision", None)
447447
torch_dtype = kwargs.pop("torch_dtype", None)

0 commit comments

Comments
 (0)