mogaio
/

Snorkel-Mistral-PairRM-DPO-Freakonomics_MTD-TCD-Lora

Inference Endpoints

Model card Files Files and versions Community

mogaio commited on Jan 29

Commit

65e5eaa

•

1 Parent(s): 981541b

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -7,7 +7,7 @@ class EndpointHandler:
     def __init__(self, path=""):
         # load model and processor from path
         base_model_name = "snorkelai/Snorkel-Mistral-PairRM-DPO"
-        lora_adaptor = "mogaio/Snorkel-Mistral-PairRM-DPO-MTD-TCD-Lora"
         self.tokenizer = AutoTokenizer.from_pretrained(base_model_name)
         self.tokenizer.pad_token = self.tokenizer.eos_token

     def __init__(self, path=""):
         # load model and processor from path
         base_model_name = "snorkelai/Snorkel-Mistral-PairRM-DPO"
+        lora_adaptor = "mogaio/Snorkel-Mistral-PairRM-DPO-Freakonomics_MTD-TCD-Lora"
         self.tokenizer = AutoTokenizer.from_pretrained(base_model_name)
         self.tokenizer.pad_token = self.tokenizer.eos_token