wagnercosta commited on
Commit
2a2e438
1 Parent(s): 1be4321

Add degug info

Browse files
Files changed (1) hide show
  1. phi3_instruct_graph.py +3 -1
phi3_instruct_graph.py CHANGED
@@ -41,7 +41,7 @@ class Phi3InstructGraph:
41
  generation_args = {
42
  "max_new_tokens": 2000,
43
  "return_full_text": False,
44
- "temperature": 0.0,
45
  "do_sample": False,
46
  }
47
 
@@ -95,4 +95,6 @@ class Phi3InstructGraph:
95
  def extract(self, text):
96
  messages = self._get_messages(text)
97
  pipe_output = self._generate(messages)
 
 
98
  return pipe_output[0]["generated_text"]
 
41
  generation_args = {
42
  "max_new_tokens": 2000,
43
  "return_full_text": False,
44
+ "temperature": 0.1,
45
  "do_sample": False,
46
  }
47
 
 
95
  def extract(self, text):
96
  messages = self._get_messages(text)
97
  pipe_output = self._generate(messages)
98
+ print("pipe_output", pipe_output)
99
+ print("pipe_output json", pipe_output[0]["generated_text"])
100
  return pipe_output[0]["generated_text"]