1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
This commit is contained in:
sayakpaul
2025-04-10 18:20:43 +05:30
parent fe176857a2
commit ff80e8a27f
2 changed files with 3 additions and 3 deletions

View File

@@ -21,8 +21,8 @@ import torch
from huggingface_hub.utils import validate_hf_hub_args
from typing_extensions import Self
from ..quantizers import DiffusersAutoQuantizer
from .. import __version__
from ..quantizers import DiffusersAutoQuantizer
from ..utils import deprecate, is_accelerate_available, logging
from .single_file_utils import (
SingleFileComponentError,

View File

@@ -405,11 +405,11 @@ def load_single_file_checkpoint(
local_files_only=None,
revision=None,
disable_mmap=False,
user_agent=None
user_agent=None,
):
if user_agent is None:
user_agent = {"file_type": "single_file", "framework": "pytorch"}
if os.path.isfile(pretrained_model_link_or_path):
pretrained_model_link_or_path = pretrained_model_link_or_path