From 9150ab02f692d61a696ca1bb0b56e133d9d888bc Mon Sep 17 00:00:00 2001 From: DN6 Date: Thu, 11 Sep 2025 08:21:57 +0530 Subject: [PATCH] update --- tests/lora/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/lora/utils.py b/tests/lora/utils.py index 625f4262ab..6c3ba945a8 100644 --- a/tests/lora/utils.py +++ b/tests/lora/utils.py @@ -167,7 +167,7 @@ class PeftLoraLoaderMixinTests: Returns the cached base pipeline output for the given scheduler. Cache is populated during setUp, so this just retrieves the value. """ - return type(self).cached_base_pipe_outs[scheduler_cls.__name__] + return self.cached_base_pipe_outs[scheduler_cls.__name__] def get_dummy_components(self, scheduler_cls=None, use_dora=False, lora_alpha=None): if self.unet_kwargs and self.transformer_kwargs: