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