Spaces:
Runtime error
Runtime error
wagnercosta
commited on
Commit
•
2a2e438
1
Parent(s):
1be4321
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"]
|