apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

remove print statements

This commit is contained in:
Sean Sube 2024-03-16 15:20:10 -05:00
parent 7b0d73664d
commit 09dfc518c1
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/prompt/compel.py
View File

@ -50,7 +50,6 @@ def wrap_encoder(text_encoder):
)
elif output_hidden_states is True:
hidden_states = [torch.from_numpy(state) for state in outputs[2:]]
print("outputs", outputs)
return SimpleNamespace(
last_hidden_state=torch.from_numpy(outputs[0]),
pooler_output=torch.from_numpy(outputs[1]),

4
api/onnx_web/prompt/grammar.py
View File

@ -181,13 +181,9 @@ class OnnxPromptVisitor(PTNodeVisitor):
]
def visit_pos_phrase(self, node, children):
print("positive phrase", len(children), children)
return parse_phrase(children, self.pos_weight)
def visit_neg_phrase(self, node, children):
print("negative phrase", len(children), children)
return parse_phrase(children, self.neg_weight)
def visit_phrase(self, node, children):