Gameur/Beta
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Voice et bot modif

Browse source
This commit is contained in:
pi 2026-06-16 17:09:34 +00:00
parent 189d56026b
commit 7333a22bcd
10774 changed files with 634644 additions and 933308 deletions
Download patch file Download diff file Expand all files Collapse all files

19
venv/lib/python3.12/site-packages/onnxruntime/quantization/calibrate.py
View file

@ -353,6 +353,14 @@ class MinMaxCalibrater(CalibraterBase):
return opset_import.version
raise RuntimeError(f"Model does not contain a version for '{op_type}'.")
def insert_nodes(tensor_name, new_nodes):
index = next(
(i for i, x in enumerate(self.model.graph.node) if tensor_name in x.input), len(self.model.graph.node)
)
for node in new_nodes:
self.model.graph.node.insert(index, node)
index += 1
def add_reduce_min_max(tensor_name, reduce_op_name):
# When doing ReduceMax/ReduceMin, ORT can't reduce on dim with value of 0 if 'keepdims' is false.
# To make the code simple, we always let keepdims to be 1.
@ -396,7 +404,7 @@ class MinMaxCalibrater(CalibraterBase):
reduce_node.input.append(reduce_axes_name)
self.model.graph.initializer.append(reduce_axes)
self.model.graph.node.extend([reduce_node, reshape_node])
insert_nodes(tensor_name, [reduce_node, reshape_node])
self.model.graph.output.append(helper.make_tensor_value_info(reduce_output, onnx_type, [None]))
for tensor in tensors:
@ -417,7 +425,14 @@ class MinMaxCalibrater(CalibraterBase):
inputs = data_reader.get_next()
if not inputs:
break
self.intermediate_outputs.append(self.infer_session.run(None, inputs))
self.intermediate_outputs.append(
[
value if sess_o.name not in self.model_original_outputs else None
for sess_o, value in zip(
self.infer_session.get_outputs(), self.infer_session.run(None, inputs), strict=False
)
]
)
if (
self.max_intermediate_outputs is not None
and len(self.intermediate_outputs) == self.max_intermediate_outputs