Update handler.py

handler.py

@@ -4,21 +4,15 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from typing import Any, Dict
 
 class EndpointHandler:
-    def __init__(
-        self,
-        path='',
-        torch_dtype=torch.bfloat16,
-        trust_remote_code=True,
-    ) -> None:
+    def __init__(self, path='', torch_dtype=torch.bfloat16, trust_remote_code=True):
+
         self.model = AutoModelForCausalLM.from_pretrained(
             path,
             torch_dtype=torch_dtype,
             trust_remote_code=trust_remote_code
         )
-        tokenizer = AutoTokenizer.from_pretrained(
-            path,
-            trust_remote_code=trust_remote_code
-        )
+        tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-20b")
+        
         if tokenizer.pad_token_id is None:
             warnings.warn(
                 "pad_token_id is not set for the tokenizer. Using eos_token_id as pad_token_id."