1

Author: zRzRzRzRzRzRzR

src/diffusers/__init__.py

@@ -289,6 +289,7 @@
             "CogVideoXVideoToVideoPipeline",
             "CogView3PlusPipeline",
             "CogView4Pipeline",
+            "CogView4ControlPipeline",
             "ConsisIDPipeline",
             "CycleDiffusionPipeline",
             "FluxControlImg2ImgPipeline",
@@ -788,6 +789,7 @@
             CogVideoXVideoToVideoPipeline,
             CogView3PlusPipeline,
             CogView4Pipeline,
+            CogView4ControlPipeline,
             ConsisIDPipeline,
             CycleDiffusionPipeline,
             FluxControlImg2ImgPipeline,

src/diffusers/models/controlnets/controlnet_cogview4.py

@@ -154,7 +154,7 @@
         "CogVideoXFunControlPipeline",
     ]
     _import_structure["cogview3"] = ["CogView3PlusPipeline"]
-    _import_structure["cogview4"] = ["CogView4Pipeline"]
+    _import_structure["cogview4"] = ["CogView4Pipeline", "CogView4ControlPipeline"]
     _import_structure["consisid"] = ["ConsisIDPipeline"]
     _import_structure["controlnet"].extend(
         [
@@ -500,7 +500,7 @@
             CogVideoXVideoToVideoPipeline,
         )
         from .cogview3 import CogView3PlusPipeline
-        from .cogview4 import CogView4Pipeline
+        from .cogview4 import CogView4Pipeline, CogView4ControlPipeline
         from .consisid import ConsisIDPipeline
         from .controlnet import (
             BlipDiffusionControlNetPipeline,

src/diffusers/pipelines/__init__.py

@@ -22,7 +22,7 @@
 from ..utils import is_sentencepiece_available
 from .aura_flow import AuraFlowPipeline
 from .cogview3 import CogView3PlusPipeline
-from .cogview4 import CogView4Pipeline
+from .cogview4 import CogView4Pipeline, CogView4ControlPipeline
 from .controlnet import (
     StableDiffusionControlNetImg2ImgPipeline,
     StableDiffusionControlNetInpaintPipeline,
@@ -140,6 +140,7 @@
         ("lumina2", Lumina2Text2ImgPipeline),
         ("cogview3", CogView3PlusPipeline),
         ("cogview4", CogView4Pipeline),
+        ("cogview4-control", CogView4ControlPipeline),
     ]
 )
 

src/diffusers/pipelines/auto_pipeline.py

@@ -23,6 +23,7 @@
     _dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
 else:
     _import_structure["pipeline_cogview4"] = ["CogView4Pipeline"]
+    _import_structure["pipeline_cogview4_control"] = ["CogView4ControlPipeline"]
 if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
     try:
         if not (is_transformers_available() and is_torch_available()):