Skip to content

Commit e187b70

Browse files
Apply suggestions from code review
Co-authored-by: Benjamin Bossan <[email protected]>
1 parent e204844 commit e187b70

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/loaders/lora_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -353,13 +353,13 @@ def _optionally_disable_offloading(cls, _pipeline):
353353
def _fetch_state_dict(cls, *args, **kwargs):
354354
deprecation_message = f"Using the `_fetch_state_dict()` method from {cls} has been deprecated and will be removed in a future version. Please use `from diffusers.loaders.lora_base import _fetch_state_dict`."
355355
deprecate("_fetch_state_dict", "0.35.0", deprecation_message)
356-
_fetch_state_dict(*args, **kwargs)
356+
return _fetch_state_dict(*args, **kwargs)
357357

358358
@classmethod
359359
def _best_guess_weight_name(cls, *args, **kwargs):
360360
deprecation_message = f"Using the `_best_guess_weight_name()` method from {cls} has been deprecated and will be removed in a future version. Please use `from diffusers.loaders.lora_base import _best_guess_weight_name`."
361361
deprecate("_best_guess_weight_name", "0.35.0", deprecation_message)
362-
_best_guess_weight_name(*args, **kwargs)
362+
return _best_guess_weight_name(*args, **kwargs)
363363

364364
def unload_lora_weights(self):
365365
"""

0 commit comments

Comments
 (0)