SITG
/

custsvc_entityextract_mistralv0.2instruct

Model card Files Files and versions

shankz7 commited on Feb 17, 2024

Commit

e07fce5

·

verified ·

1 Parent(s): 31f8b30

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -16,7 +16,7 @@ class EndpointHandler():
         tokenizer.pad_token = tokenizer.eos_token
         tokenizer.padding_side = "right"
-        config = PeftConfig.from_pretrained(".")
         model = PeftModel.from_pretrained(model, config)

         tokenizer.pad_token = tokenizer.eos_token
         tokenizer.padding_side = "right"
+        config = PeftConfig.from_pretrained("")
         model = PeftModel.from_pretrained(model, config)