Skip to content

GitLab

Unverified Commit 67f5cce2 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix autopipeline for kolors img2img (#9212) 

fix
parent d72bbc68
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/pipelines/auto_pipeline.py
View file @ 67f5cce2
......@@ -163,12 +163,12 @@ _AUTO_INPAINT_DECODER_PIPELINES_MAPPING = OrderedDict(
)
if is_sentencepiece_available():
from .kolors import KolorsPipeline
from .kolors import KolorsImg2ImgPipeline, KolorsPipeline
from .pag import KolorsPAGPipeline
AUTO_TEXT2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsPipeline
AUTO_TEXT2IMAGE_PIPELINES_MAPPING["kolors-pag"] = KolorsPAGPipeline
AUTO_IMAGE2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsPipeline
AUTO_IMAGE2IMAGE_PIPELINES_MAPPING["kolors"] = KolorsImg2ImgPipeline
SUPPORTED_TASKS_MAPPINGS = [
AUTO_TEXT2IMAGE_PIPELINES_MAPPING,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment