public/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-11-29 14:39:48 +09:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

add comment about gross code

Browse Source
This commit is contained in:
mertalev
2024-02-02 22:18:14 -05:00
parent b374052de2
commit 1ed748e7f3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
machine-learning/app/models/base.py
View File

@@ -143,6 +143,7 @@ class InferenceModel(ABC):
inputs = self._get_static_dims(inferred.graph.input) inputs = self._get_static_dims(inferred.graph.input)
outputs = self._get_static_dims(inferred.graph.output) outputs = self._get_static_dims(inferred.graph.output)
# check_model gets called in update_inputs_outputs_dims and doesn't work for large models
check_model = onnx.checker.check_model check_model = onnx.checker.check_model
try: try:
onnx.checker.check_model = lambda _: None onnx.checker.check_model = lambda _: None