oops typo

approximatelabs · May 19, 2023 · ee8f01d · ee8f01d
1 parent 2029ef6
commit ee8f01d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lambdaprompt/backends.py b/lambdaprompt/backends.py
@@ -225,7 +225,7 @@ def __init__(self, model_name, torch_dtype=None, trust_remote_code=True, use_aut
         torch_dtype = torch_dtype or torch.bfloat16
         super().__init__(**param_override)
         config = AutoConfig.from_pretrained(
-            "HuggingFaceH4/starchat-alpha",
+            model_name,
             trust_remote_code=True
         )
         self.model = AutoModelForCausalLM.from_pretrained(