mixins, offload_seq, option_components

Author: hlky

src/diffusers/models/transformers/transformer_hidream_image.py

@@ -6,7 +6,7 @@
 import torch.nn.functional as F
 
 from ...configuration_utils import ConfigMixin, register_to_config
-from ...loaders import FromOriginalModelMixin, PeftAdapterMixin
+from ...loaders import PeftAdapterMixin
 from ...models.modeling_outputs import Transformer2DModelOutput
 from ...models.modeling_utils import ModelMixin
 from ...utils import USE_PEFT_BACKEND, is_torch_version, logging, scale_lora_layers, unscale_lora_layers
@@ -656,7 +656,7 @@ def forward(
         )
 
 
-class HiDreamImageTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin):
+class HiDreamImageTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
     _supports_gradient_checkpointing = True
     _no_split_modules = ["HiDreamImageBlock"]
 

src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py

@@ -13,7 +13,6 @@
 )
 
 from ...image_processor import VaeImageProcessor
-from ...loaders import FromSingleFileMixin
 from ...models import AutoencoderKL, HiDreamImageTransformer2DModel
 from ...pipelines.pipeline_utils import DiffusionPipeline
 from ...schedulers import FlowMatchEulerDiscreteScheduler, UniPCMultistepScheduler
@@ -109,11 +108,8 @@ def retrieve_timesteps(
     return timesteps, num_inference_steps
 
 
-class HiDreamImagePipeline(DiffusionPipeline, FromSingleFileMixin):
-    model_cpu_offload_seq = (
-        "text_encoder->text_encoder_2->text_encoder_3->text_encoder_4->image_encoder->transformer->vae"
-    )
-    _optional_components = ["image_encoder", "feature_extractor"]
+class HiDreamImagePipeline(DiffusionPipeline):
+    model_cpu_offload_seq = "text_encoder->text_encoder_2->text_encoder_3->text_encoder_4->transformer->vae"
     _callback_tensor_inputs = ["latents", "prompt_embeds"]
 
     def __init__(