made it more pythonic

EleutherAI · Jan 6, 2025 · 403fd58 · 403fd58
1 parent aa3c937
commit 403fd58
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lm_eval/models/huggingface.py b/lm_eval/models/huggingface.py
@@ -637,8 +637,7 @@ def _create_model(
 
             # check if peft contains peft revision according to "adapter_id@revision" format (f.e. used in [TGI endpoints](https://huggingface.co/docs/text-generation-inference/main/en/conceptual/lora#specifying-lora-models))
             if "@" in peft:
-                peft = peft.split("@")[0]
-                peft_revision = peft.split("@")[1]
+                peft, peft_revision = peft.split("@", 1)
                 self._model = PeftModel.from_pretrained(
                     self._model, peft, revision=peft_revision
                 )