diff --git a/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py b/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py index e16dedb53674..e22441c6cde7 100644 --- a/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py +++ b/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py @@ -36,7 +36,7 @@ ```py >>> import torch >>> from transformers import PreTrainedTokenizerFast, LlamaForCausalLM - >>> from diffusers import UniPCMultistepScheduler, HiDreamImagePipeline, HiDreamImageTransformer2DModel + >>> from diffusers import UniPCMultistepScheduler, HiDreamImagePipeline >>> scheduler = UniPCMultistepScheduler( ... flow_shift=3.0, prediction_type="flow_prediction", use_flow_sigmas=True @@ -50,16 +50,11 @@ ... torch_dtype=torch.bfloat16, ... ) - >>> transformer = HiDreamImageTransformer2DModel.from_pretrained( - ... "HiDream-ai/HiDream-I1-Full", subfolder="transformer", torch_dtype=torch.bfloat16 - ... ) - >>> pipe = HiDreamImagePipeline.from_pretrained( ... "HiDream-ai/HiDream-I1-Full", ... scheduler=scheduler, ... tokenizer_4=tokenizer_4, ... text_encoder_4=text_encoder_4, - ... transformer=transformer, ... torch_dtype=torch.bfloat16, ... ) >>> pipe.enable_model_cpu_offload()