fix imports

a-r-r-o-w · a-r-r-o-w · commit 98f102385b10 · 2024-09-12T22:18:44.000+02:00
diff --git a/src/diffusers/pipelines/__init__.py b/src/diffusers/pipelines/__init__.py
@@ -465,7 +465,7 @@
         )
         from .aura_flow import AuraFlowPipeline
         from .blip_diffusion import BlipDiffusionPipeline
-        from .cogvideo import CogVideoXPipeline, CogVideoXVideoToVideoPipeline
+        from .cogvideo import CogVideoXPipeline, CogVideoXImageToVideoPipeline, CogVideoXVideoToVideoPipeline
         from .controlnet import (
             BlipDiffusionControlNetPipeline,
             StableDiffusionControlNetImg2ImgPipeline,

Original file line number	Diff line number	Diff line change
`@@ -465,7 +465,7 @@`
`465`	`465`	`)`
`466`	`466`	`from .aura_flow import AuraFlowPipeline`
`467`	`467`	`from .blip_diffusion import BlipDiffusionPipeline`
`468`		`- from .cogvideo import CogVideoXPipeline, CogVideoXVideoToVideoPipeline`
	`468`	`+ from .cogvideo import CogVideoXPipeline, CogVideoXImageToVideoPipeline, CogVideoXVideoToVideoPipeline`
`469`	`469`	`from .controlnet import (`
`470`	`470`	`BlipDiffusionControlNetPipeline,`
`471`	`471`	`StableDiffusionControlNetImg2ImgPipeline,`