mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
quality
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user