Update handler.py

handler.py

@@ -19,7 +19,7 @@ class EndpointHandler():
         if len(inputs) == 0:
             raise ValueError("prompt cannot be empty")
         model_input = self.tokenizer(inputs, return_tensors="pt").to(self.device_map)
-        output = self.llm_model.generate(input_ids=model_input["input_ids"].to(self.device_map),
+        output = self.model.generate(input_ids=model_input["input_ids"].to(self.device_map),
                                   use_cache=False,
                                   temperature=0.1, top_k=1, top_p=1.0, repetition_penalty=1.4,
                                   max_new_tokens=256,