Spaces:
Runtime error
Runtime error
# Copyright 2022 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 .deprecation_utils import deprecate | |
from .import_utils import ( | |
ENV_VARS_TRUE_AND_AUTO_VALUES, | |
ENV_VARS_TRUE_VALUES, | |
USE_JAX, | |
USE_TF, | |
USE_TORCH, | |
DummyObject, | |
is_accelerate_available, | |
is_flax_available, | |
is_inflect_available, | |
is_modelcards_available, | |
is_onnx_available, | |
is_safetensors_available, | |
is_scipy_available, | |
is_tf_available, | |
is_torch_available, | |
is_torch_version, | |
is_transformers_available, | |
is_transformers_version, | |
is_unidecode_available, | |
requires_backends, | |
) | |
from .logging import get_logger | |
from .outputs import BaseOutput | |
from .pil_utils import PIL_INTERPOLATION | |
if is_torch_available(): | |
from .testing_utils import ( | |
floats_tensor, | |
load_hf_numpy, | |
load_image, | |
load_numpy, | |
parse_flag_from_env, | |
require_torch_gpu, | |
slow, | |
torch_all_close, | |
torch_device, | |
) | |
logger = get_logger(__name__) | |
hf_cache_home = os.path.expanduser( | |
os.getenv("HF_HOME", os.path.join(os.getenv("XDG_CACHE_HOME", "~/.cache"), "huggingface")) | |
) | |
default_cache_path = os.path.join(hf_cache_home, "diffusers") | |
CONFIG_NAME = "config.json" | |
WEIGHTS_NAME = "diffusion_pytorch_model.bin" | |
FLAX_WEIGHTS_NAME = "diffusion_flax_model.msgpack" | |
ONNX_WEIGHTS_NAME = "model.onnx" | |
SAFETENSORS_WEIGHTS_NAME = "diffusion_pytorch_model.safetensors" | |
ONNX_EXTERNAL_WEIGHTS_NAME = "weights.pb" | |
HUGGINGFACE_CO_RESOLVE_ENDPOINT = "https://huggingface.co" | |
DIFFUSERS_CACHE = default_cache_path | |
DIFFUSERS_DYNAMIC_MODULE_NAME = "diffusers_modules" | |
HF_MODULES_CACHE = os.getenv("HF_MODULES_CACHE", os.path.join(hf_cache_home, "modules")) | |
_COMPATIBLE_STABLE_DIFFUSION_SCHEDULERS = [ | |
"DDIMScheduler", | |
"DDPMScheduler", | |
"PNDMScheduler", | |
"LMSDiscreteScheduler", | |
"EulerDiscreteScheduler", | |
"HeunDiscreteScheduler", | |
"EulerAncestralDiscreteScheduler", | |
"DPMSolverMultistepScheduler", | |
] | |