diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py index 1acb4494e1..27bbd35016 100644 --- a/src/diffusers/__init__.py +++ b/src/diffusers/__init__.py @@ -353,8 +353,8 @@ else: "AuraFlowPipeline", "BlipDiffusionControlNetPipeline", "BlipDiffusionPipeline", - "CLIPImageProjection", "ChromaPipeline", + "CLIPImageProjection", "CogVideoXFunControlPipeline", "CogVideoXImageToVideoPipeline", "CogVideoXPipeline", diff --git a/src/diffusers/pipelines/chroma/pipeline_chroma.py b/src/diffusers/pipelines/chroma/pipeline_chroma.py index 2b1b516ffc..564faad838 100644 --- a/src/diffusers/pipelines/chroma/pipeline_chroma.py +++ b/src/diffusers/pipelines/chroma/pipeline_chroma.py @@ -57,7 +57,9 @@ EXAMPLE_DOC_STRING = """ >>> import torch >>> from diffusers import ChromaPipeline - >>> pipe = ChromaPipeline.from_single_file("chroma-unlocked-v35-detail-calibrated.safetensors", torch_dtype=torch.bfloat16) + >>> pipe = ChromaPipeline.from_single_file( + ... "chroma-unlocked-v35-detail-calibrated.safetensors", torch_dtype=torch.bfloat16 + ... ) >>> pipe.to("cuda") >>> prompt = "A cat holding a sign that says hello world" >>> image = pipe(prompt, num_inference_steps=28, guidance_scale=4.0).images[0] @@ -630,9 +632,9 @@ class ChromaPipeline( Examples: Returns: - [`~pipelines.chroma.ChromaPipelineOutput`] or `tuple`: [`~pipelines.chroma.ChromaPipelineOutput`] if `return_dict` - is True, otherwise a `tuple`. When returning a tuple, the first element is a list with the generated - images. + [`~pipelines.chroma.ChromaPipelineOutput`] or `tuple`: [`~pipelines.chroma.ChromaPipelineOutput`] if + `return_dict` is True, otherwise a `tuple`. When returning a tuple, the first element is a list with the + generated images. """ height = height or self.default_sample_size * self.vae_scale_factor