Add degug info
Browse files- 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.
|
| 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"]
|