poiccard commited on
Commit
24f4d6c
·
1 Parent(s): 344dbbf

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -0
handler.py CHANGED
@@ -21,6 +21,7 @@ class EndpointHandler():
21
  # text = self.prompt_ar.format_map({'Question':inputs})
22
  # the prompt will be preapred on the backend fully
23
  text = inputs
 
24
 
25
  input_ids = self.tokenizer(text, return_tensors="pt").input_ids
26
  inputs = input_ids.to(self.device)
 
21
  # text = self.prompt_ar.format_map({'Question':inputs})
22
  # the prompt will be preapred on the backend fully
23
  text = inputs
24
+ print(f"///prompt: text")
25
 
26
  input_ids = self.tokenizer(text, return_tensors="pt").input_ids
27
  inputs = input_ids.to(self.device)