Skip to content

Commit 67f5cce

Browse files
authored
fix autopipeline for kolors img2img (#9212)
fix
1 parent d72bbc6 commit 67f5cce

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/auto_pipeline.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,12 +163,12 @@
163163
)
164164

165165
if is_sentencepiece_available():
166-
from .kolors import KolorsPipeline
166+
from .kolors import KolorsImg2ImgPipeline, KolorsPipeline
167167
from .pag import KolorsPAGPipeline
168168

169169
AUTO_TEXT2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsPipeline
170170
AUTO_TEXT2IMAGE_PIPELINES_MAPPING["kolors-pag"] = KolorsPAGPipeline
171-
AUTO_IMAGE2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsPipeline
171+
AUTO_IMAGE2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsImg2ImgPipeline
172172

173173
SUPPORTED_TASKS_MAPPINGS = [
174174
AUTO_TEXT2IMAGE_PIPELINES_MAPPING,

0 commit comments

Comments
 (0)