We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d72bbc6 commit 67f5cceCopy full SHA for 67f5cce
src/diffusers/pipelines/auto_pipeline.py
@@ -163,12 +163,12 @@
163
)
164
165
if is_sentencepiece_available():
166
- from .kolors import KolorsPipeline
+ from .kolors import KolorsImg2ImgPipeline, KolorsPipeline
167
from .pag import KolorsPAGPipeline
168
169
AUTO_TEXT2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsPipeline
170
AUTO_TEXT2IMAGE_PIPELINES_MAPPING["kolors-pag"] = KolorsPAGPipeline
171
- AUTO_IMAGE2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsPipeline
+ AUTO_IMAGE2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsImg2ImgPipeline
172
173
SUPPORTED_TASKS_MAPPINGS = [
174
AUTO_TEXT2IMAGE_PIPELINES_MAPPING,
0 commit comments