From de8f9b2528988136f1bdf54113847de300c4b413 Mon Sep 17 00:00:00 2001 From: Noah Yoshida Date: Wed, 3 Apr 2024 11:33:24 -0700 Subject: [PATCH] cleanup --- server/lorax_server/adapters/weights.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/lorax_server/adapters/weights.py b/server/lorax_server/adapters/weights.py index 8055e1022..3ea47e51f 100644 --- a/server/lorax_server/adapters/weights.py +++ b/server/lorax_server/adapters/weights.py @@ -106,7 +106,7 @@ def ranks(self) -> Set[int]: return set( rank_data.rank for layer_data in self.data.values() - for rank_data in layer_data.get(LORA, []).rank_data.values() if LORA in layer_data + for rank_data in layer_data.get(LORA).rank_data.values() if LORA in layer_data ) @property