codys12
/

Mistral-7b-Pathway-128k-3

Model card Files Files and versions

codys12 commited on Nov 17, 2023

Commit

dd7e51e

·

1 Parent(s): d38a401

Upload handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -45,8 +45,8 @@ class EndpointHandler():
         # Call the generate function
         output = generate(
-            tokenizer=tokenizer,
-            model=model,
             message=message,
             chat_history=chat_history,
             system_prompt=system_prompt,

         # Call the generate function
         output = generate(
+            tokenizer=self.tokenizer,
+            model=self.model,
             message=message,
             chat_history=chat_history,
             system_prompt=system_prompt,