add tests

Author: sayakpaul

src/diffusers/loaders/lora_base.py

@@ -349,6 +349,18 @@ def _optionally_disable_offloading(cls, _pipeline):
 
         return (is_model_cpu_offload, is_sequential_cpu_offload)
 
+    @classmethod
+    def _fetch_state_dict(cls, *args, **kwargs):
+        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`."
+        deprecate("_fetch_state_dict", "0.35.0", deprecation_message)
+        _fetch_state_dict(*args, **kwargs)
+
+    @classmethod
+    def _best_guess_weight_name(cls, *args, **kwargs):
+        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`."
+        deprecate("_best_guess_weight_name", "0.35.0", deprecation_message)
+        _best_guess_weight_name(*args, **kwargs)
+
     def unload_lora_weights(self):
         """
         Unloads the LoRA parameters.

tests/lora/test_deprecated_utilities.py

@@ -0,0 +1,39 @@
+import os
+import tempfile
+import unittest
+
+import torch
+
+from diffusers.loaders.lora_base import LoraBaseMixin
+
+
+class UtilityMethodDeprecationTests(unittest.TestCase):
+    def test_fetch_state_dict_cls_method_raises_warning(self):
+        state_dict = torch.nn.Linear(3, 3).state_dict()
+        with self.assertWarns(FutureWarning) as warning:
+            _ = LoraBaseMixin._fetch_state_dict(
+                state_dict,
+                weight_name=None,
+                use_safetensors=False,
+                local_files_only=True,
+                cache_dir=None,
+                force_download=False,
+                proxies=None,
+                token=None,
+                revision=None,
+                subfolder=None,
+                user_agent=None,
+                allow_pickle=None,
+            )
+        warning_message = str(warning.warnings[0].message)
+        assert "Using the `_fetch_state_dict()` method from" in warning_message
+
+    def test_best_guess_weight_name_cls_method_raises_warning(self):
+        with tempfile.TemporaryDirectory() as tmpdir:
+            state_dict = torch.nn.Linear(3, 3).state_dict()
+            torch.save(state_dict, os.path.join(tmpdir, "pytorch_lora_weights.bin"))
+
+            with self.assertWarns(FutureWarning) as warning:
+                _ = LoraBaseMixin._best_guess_weight_name(pretrained_model_name_or_path_or_dict=tmpdir)
+            warning_message = str(warning.warnings[0].message)
+            assert "Using the `_best_guess_weight_name()` method from" in warning_message