Compare commits

..

4 Commits

Author SHA1 Message Date
Dhruv Nair
16fe810b3e test 2023-11-07 05:35:25 +00:00
Dhruv Nair
79e7fb3a16 test 2023-11-07 05:33:25 +00:00
Dhruv Nair
c15d9936c1 test 2023-11-07 05:27:34 +00:00
Beinsezii
9bafef34bd Add Pixart to AUTO_TEXT2IMAGE_PIPELINES_MAPPING (#5664) 2023-11-07 07:45:56 +05:30
4 changed files with 6 additions and 3 deletions

View File

@@ -28,5 +28,6 @@ jobs:
pip install pytest
- name: Check for soft dependencies
run: |
python utils/print_env.py
python -c "from diffusers import UNet2DConditionModel; print(type(UNet2DConditionModel))"
pytest tests/others/test_dependencies.py

View File

@@ -97,7 +97,7 @@ _deps = [
"filelock",
"flax>=0.4.1",
"hf-doc-builder>=0.3.0",
"huggingface-hub==v0.19.0.rc0",
"huggingface-hub>=0.13.2",
"requests-mock==1.10.0",
"importlib_metadata",
"invisible-watermark>=0.2.0",

View File

@@ -10,7 +10,7 @@ deps = {
"filelock": "filelock",
"flax": "flax>=0.4.1",
"hf-doc-builder": "hf-doc-builder>=0.3.0",
"huggingface-hub": "huggingface-hub==v0.19.0.rc0",
"huggingface-hub": "huggingface-hub>=0.13.2",
"requests-mock": "requests-mock==1.10.0",
"importlib_metadata": "importlib_metadata",
"invisible-watermark": "invisible-watermark>=0.2.0",

View File

@@ -43,6 +43,7 @@ from .kandinsky2_2 import (
KandinskyV22Pipeline,
)
from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
from .pixart_alpha import PixArtAlphaPipeline
from .stable_diffusion import (
StableDiffusionImg2ImgPipeline,
StableDiffusionInpaintPipeline,
@@ -67,6 +68,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline),
("wuerstchen", WuerstchenCombinedPipeline),
("lcm", LatentConsistencyModelPipeline),
("pixart", PixArtAlphaPipeline),
]
)