mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-12 23:44:30 +08:00
Compare commits
4 Commits
custom-cod
...
v0.8.1
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3876b20b5f | ||
|
|
9fee706d9b | ||
|
|
443fcc25d7 | ||
|
|
2a49278a51 |
2
setup.py
2
setup.py
@@ -212,7 +212,7 @@ install_requires = [
|
|||||||
|
|
||||||
setup(
|
setup(
|
||||||
name="diffusers",
|
name="diffusers",
|
||||||
version="0.8.0", # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
|
version="0.8.1", # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
|
||||||
description="Diffusers",
|
description="Diffusers",
|
||||||
long_description=open("README.md", "r", encoding="utf-8").read(),
|
long_description=open("README.md", "r", encoding="utf-8").read(),
|
||||||
long_description_content_type="text/markdown",
|
long_description_content_type="text/markdown",
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from .utils import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
__version__ = "0.8.0"
|
__version__ = "0.8.1"
|
||||||
|
|
||||||
from .configuration_utils import ConfigMixin
|
from .configuration_utils import ConfigMixin
|
||||||
from .onnx_utils import OnnxRuntimeModel
|
from .onnx_utils import OnnxRuntimeModel
|
||||||
|
|||||||
@@ -6,7 +6,14 @@ import numpy as np
|
|||||||
import PIL
|
import PIL
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
|
|
||||||
from ...utils import BaseOutput, is_flax_available, is_onnx_available, is_torch_available, is_transformers_available
|
from ...utils import (
|
||||||
|
BaseOutput,
|
||||||
|
is_flax_available,
|
||||||
|
is_onnx_available,
|
||||||
|
is_torch_available,
|
||||||
|
is_transformers_available,
|
||||||
|
is_transformers_version,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
@@ -30,12 +37,16 @@ class StableDiffusionPipelineOutput(BaseOutput):
|
|||||||
if is_transformers_available() and is_torch_available():
|
if is_transformers_available() and is_torch_available():
|
||||||
from .pipeline_cycle_diffusion import CycleDiffusionPipeline
|
from .pipeline_cycle_diffusion import CycleDiffusionPipeline
|
||||||
from .pipeline_stable_diffusion import StableDiffusionPipeline
|
from .pipeline_stable_diffusion import StableDiffusionPipeline
|
||||||
from .pipeline_stable_diffusion_image_variation import StableDiffusionImageVariationPipeline
|
|
||||||
from .pipeline_stable_diffusion_img2img import StableDiffusionImg2ImgPipeline
|
from .pipeline_stable_diffusion_img2img import StableDiffusionImg2ImgPipeline
|
||||||
from .pipeline_stable_diffusion_inpaint import StableDiffusionInpaintPipeline
|
from .pipeline_stable_diffusion_inpaint import StableDiffusionInpaintPipeline
|
||||||
from .pipeline_stable_diffusion_inpaint_legacy import StableDiffusionInpaintPipelineLegacy
|
from .pipeline_stable_diffusion_inpaint_legacy import StableDiffusionInpaintPipelineLegacy
|
||||||
from .safety_checker import StableDiffusionSafetyChecker
|
from .safety_checker import StableDiffusionSafetyChecker
|
||||||
|
|
||||||
|
if is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0.dev0"):
|
||||||
|
from .pipeline_stable_diffusion_image_variation import StableDiffusionImageVariationPipeline
|
||||||
|
else:
|
||||||
|
from ...utils.dummy_torch_and_transformers_objects import StableDiffusionImageVariationPipeline
|
||||||
|
|
||||||
if is_transformers_available() and is_onnx_available():
|
if is_transformers_available() and is_onnx_available():
|
||||||
from .pipeline_onnx_stable_diffusion import OnnxStableDiffusionPipeline, StableDiffusionOnnxPipeline
|
from .pipeline_onnx_stable_diffusion import OnnxStableDiffusionPipeline, StableDiffusionOnnxPipeline
|
||||||
from .pipeline_onnx_stable_diffusion_img2img import OnnxStableDiffusionImg2ImgPipeline
|
from .pipeline_onnx_stable_diffusion_img2img import OnnxStableDiffusionImg2ImgPipeline
|
||||||
|
|||||||
@@ -1,9 +1,16 @@
|
|||||||
from ...utils import is_torch_available, is_transformers_available
|
from ...utils import is_torch_available, is_transformers_available, is_transformers_version
|
||||||
|
|
||||||
|
|
||||||
if is_transformers_available() and is_torch_available():
|
if is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0.dev0"):
|
||||||
from .modeling_text_unet import UNetFlatConditionModel
|
from .modeling_text_unet import UNetFlatConditionModel
|
||||||
from .pipeline_versatile_diffusion import VersatileDiffusionPipeline
|
from .pipeline_versatile_diffusion import VersatileDiffusionPipeline
|
||||||
from .pipeline_versatile_diffusion_dual_guided import VersatileDiffusionDualGuidedPipeline
|
from .pipeline_versatile_diffusion_dual_guided import VersatileDiffusionDualGuidedPipeline
|
||||||
from .pipeline_versatile_diffusion_image_variation import VersatileDiffusionImageVariationPipeline
|
from .pipeline_versatile_diffusion_image_variation import VersatileDiffusionImageVariationPipeline
|
||||||
from .pipeline_versatile_diffusion_text_to_image import VersatileDiffusionTextToImagePipeline
|
from .pipeline_versatile_diffusion_text_to_image import VersatileDiffusionTextToImagePipeline
|
||||||
|
else:
|
||||||
|
from ...utils.dummy_torch_and_transformers_objects import (
|
||||||
|
VersatileDiffusionDualGuidedPipeline,
|
||||||
|
VersatileDiffusionImageVariationPipeline,
|
||||||
|
VersatileDiffusionPipeline,
|
||||||
|
VersatileDiffusionTextToImagePipeline,
|
||||||
|
)
|
||||||
|
|||||||
@@ -33,6 +33,7 @@ from .import_utils import (
|
|||||||
is_torch_available,
|
is_torch_available,
|
||||||
is_torch_version,
|
is_torch_version,
|
||||||
is_transformers_available,
|
is_transformers_available,
|
||||||
|
is_transformers_version,
|
||||||
is_unidecode_available,
|
is_unidecode_available,
|
||||||
requires_backends,
|
requires_backends,
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -303,6 +303,17 @@ def requires_backends(obj, backends):
|
|||||||
if failed:
|
if failed:
|
||||||
raise ImportError("".join(failed))
|
raise ImportError("".join(failed))
|
||||||
|
|
||||||
|
if name in [
|
||||||
|
"VersatileDiffusionTextToImagePipeline",
|
||||||
|
"VersatileDiffusionPipeline",
|
||||||
|
"VersatileDiffusionDualGuidedPipeline",
|
||||||
|
"StableDiffusionImageVariationPipeline",
|
||||||
|
] and is_transformers_version("<", "4.25.0.dev0"):
|
||||||
|
raise ImportError(
|
||||||
|
f"You need to install `transformers` from 'main' in order to use {name}: \n```\n pip install"
|
||||||
|
" git+https://github.com/huggingface/transformers \n```"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class DummyObject(type):
|
class DummyObject(type):
|
||||||
"""
|
"""
|
||||||
@@ -347,3 +358,17 @@ def is_torch_version(operation: str, version: str):
|
|||||||
A string version of PyTorch
|
A string version of PyTorch
|
||||||
"""
|
"""
|
||||||
return compare_versions(parse(_torch_version), operation, version)
|
return compare_versions(parse(_torch_version), operation, version)
|
||||||
|
|
||||||
|
|
||||||
|
def is_transformers_version(operation: str, version: str):
|
||||||
|
"""
|
||||||
|
Args:
|
||||||
|
Compares the current Transformers version to a given reference with an operation.
|
||||||
|
operation (`str`):
|
||||||
|
A string representation of an operator, such as `">"` or `"<="`
|
||||||
|
version (`str`):
|
||||||
|
A string version of PyTorch
|
||||||
|
"""
|
||||||
|
if not _transformers_available:
|
||||||
|
return False
|
||||||
|
return compare_versions(parse(_transformers_version), operation, version)
|
||||||
|
|||||||
Reference in New Issue
Block a user