using next

Signed-off-by: wang.yuqi <[email protected]>

Author: noooop

vllm/model_executor/models/adapters.py

@@ -439,9 +439,9 @@ def load_weights_using_from_2_way_softmax(
 
     # ModelForPooling is dynamically defined inside the _create_pooling_model_cls
     # function, so we need use this hacky method to obtain it.
-    pooling_model_cls = [
+    pooling_model_cls = next(
         x for x in type(model).__mro__ if x.__name__ == "ModelForPooling"
-    ][0]
+    )
     loaded_weights = pooling_model_cls.load_weights(model, weights, load_lm_head=True)
 
     from vllm.transformers_utils.tokenizer import get_tokenizer