diff --git a/src/diffusers/loaders/lora/lora_pipeline.py b/src/diffusers/loaders/lora/lora_pipeline.py index 75613bfcb4..32e5fa1e6b 100644 --- a/src/diffusers/loaders/lora/lora_pipeline.py +++ b/src/diffusers/loaders/lora/lora_pipeline.py @@ -5739,7 +5739,7 @@ class HiDreamImageLoraLoaderMixin(LoraBaseMixin): @classmethod @validate_hf_hub_args - # Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.lora_state_dict + # Copied from diffusers.loaders.lora.lora_pipeline.CogVideoXLoraLoaderMixin.lora_state_dict def lora_state_dict( cls, pretrained_model_name_or_path_or_dict: Union[str, Dict[str, torch.Tensor]], @@ -5835,7 +5835,7 @@ class HiDreamImageLoraLoaderMixin(LoraBaseMixin): return state_dict - # Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.load_lora_weights + # Copied from diffusers.loaders.lora.lora_pipeline.CogVideoXLoraLoaderMixin.load_lora_weights def load_lora_weights( self, pretrained_model_name_or_path_or_dict: Union[str, Dict[str, torch.Tensor]], @@ -5894,7 +5894,7 @@ class HiDreamImageLoraLoaderMixin(LoraBaseMixin): ) @classmethod - # Copied from diffusers.loaders.lora_pipeline.SD3LoraLoaderMixin.load_lora_into_transformer with SD3Transformer2DModel->HiDreamImageTransformer2DModel + # Copied from diffusers.loaders.lora.lora_pipeline.SD3LoraLoaderMixin.load_lora_into_transformer with SD3Transformer2DModel->HiDreamImageTransformer2DModel def load_lora_into_transformer( cls, state_dict, transformer, adapter_name=None, _pipeline=None, low_cpu_mem_usage=False, hotswap: bool = False ): @@ -5934,7 +5934,7 @@ class HiDreamImageLoraLoaderMixin(LoraBaseMixin): ) @classmethod - # Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.save_lora_weights + # Copied from diffusers.loaders.lora.lora_pipeline.CogVideoXLoraLoaderMixin.save_lora_weights def save_lora_weights( cls, save_directory: Union[str, os.PathLike], @@ -5981,7 +5981,7 @@ class HiDreamImageLoraLoaderMixin(LoraBaseMixin): safe_serialization=safe_serialization, ) - # Copied from diffusers.loaders.lora_pipeline.SanaLoraLoaderMixin.fuse_lora + # Copied from diffusers.loaders.lora.lora_pipeline.SanaLoraLoaderMixin.fuse_lora def fuse_lora( self, components: List[str] = ["transformer"], @@ -6029,7 +6029,7 @@ class HiDreamImageLoraLoaderMixin(LoraBaseMixin): **kwargs, ) - # Copied from diffusers.loaders.lora_pipeline.SanaLoraLoaderMixin.unfuse_lora + # Copied from diffusers.loaders.lora.lora_pipeline.SanaLoraLoaderMixin.unfuse_lora def unfuse_lora(self, components: List[str] = ["transformer"], **kwargs): r""" Reverses the effect of