Spaces:
Running
on
Zero
Running
on
Zero
# Copyright 2023 The HuggingFace Inc. team. All rights reserved. | |
# | |
# Licensed under the Apache License, Version 2.0 (the "License"); | |
# you may not use this file except in compliance with the License. | |
# You may obtain a copy of the License at | |
# | |
# http://www.apache.org/licenses/LICENSE-2.0 | |
# | |
# Unless required by applicable law or agreed to in writing, software | |
# distributed under the License is distributed on an "AS IS" BASIS, | |
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
# See the License for the specific language governing permissions and | |
# limitations under the License. | |
import os | |
from packaging import version | |
from .. import __version__ | |
from .constants import ( | |
CONFIG_NAME, | |
DEPRECATED_REVISION_ARGS, | |
DIFFUSERS_DYNAMIC_MODULE_NAME, | |
FLAX_WEIGHTS_NAME, | |
HF_MODULES_CACHE, | |
HUGGINGFACE_CO_RESOLVE_ENDPOINT, | |
MIN_PEFT_VERSION, | |
ONNX_EXTERNAL_WEIGHTS_NAME, | |
ONNX_WEIGHTS_NAME, | |
SAFETENSORS_WEIGHTS_NAME, | |
USE_PEFT_BACKEND, | |
WEIGHTS_NAME, | |
) | |
from .deprecation_utils import deprecate | |
from .doc_utils import replace_example_docstring | |
from .dynamic_modules_utils import get_class_from_dynamic_module | |
from .export_utils import export_to_gif, export_to_obj, export_to_ply, export_to_video | |
from .hub_utils import ( | |
PushToHubMixin, | |
_add_variant, | |
_get_model_file, | |
extract_commit_hash, | |
http_user_agent, | |
) | |
from .import_utils import ( | |
BACKENDS_MAPPING, | |
DIFFUSERS_SLOW_IMPORT, | |
ENV_VARS_TRUE_AND_AUTO_VALUES, | |
ENV_VARS_TRUE_VALUES, | |
USE_JAX, | |
USE_TF, | |
USE_TORCH, | |
DummyObject, | |
OptionalDependencyNotAvailable, | |
_LazyModule, | |
get_objects_from_module, | |
is_accelerate_available, | |
is_accelerate_version, | |
is_bs4_available, | |
is_flax_available, | |
is_ftfy_available, | |
is_inflect_available, | |
is_invisible_watermark_available, | |
is_k_diffusion_available, | |
is_k_diffusion_version, | |
is_librosa_available, | |
is_note_seq_available, | |
is_onnx_available, | |
is_peft_available, | |
is_scipy_available, | |
is_tensorboard_available, | |
is_torch_available, | |
is_torch_version, | |
is_torch_xla_available, | |
is_torchsde_available, | |
is_transformers_available, | |
is_transformers_version, | |
is_unidecode_available, | |
is_wandb_available, | |
is_xformers_available, | |
requires_backends, | |
) | |
from .loading_utils import load_image | |
from .logging import get_logger | |
from .outputs import BaseOutput | |
from .peft_utils import ( | |
check_peft_version, | |
delete_adapter_layers, | |
get_adapter_name, | |
get_peft_kwargs, | |
recurse_remove_peft_layers, | |
scale_lora_layers, | |
set_adapter_layers, | |
set_weights_and_activate_adapters, | |
unscale_lora_layers, | |
) | |
from .pil_utils import PIL_INTERPOLATION, make_image_grid, numpy_to_pil, pt_to_pil | |
from .state_dict_utils import ( | |
convert_all_state_dict_to_peft, | |
convert_state_dict_to_diffusers, | |
convert_state_dict_to_kohya, | |
convert_state_dict_to_peft, | |
convert_unet_state_dict_to_peft, | |
) | |
logger = get_logger(__name__) | |
def check_min_version(min_version): | |
if version.parse(__version__) < version.parse(min_version): | |
if "dev" in min_version: | |
error_message = ( | |
"This example requires a source install from HuggingFace diffusers (see " | |
"`https://huggingface.co/docs/diffusers/installation#install-from-source`)," | |
) | |
else: | |
error_message = f"This example requires a minimum version of {min_version}," | |
error_message += f" but the version found is {__version__}.\n" | |
raise ImportError(error_message) | |