self-hosted/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-01-12 15:32:36 +02:00
Code Issues Releases Activity

formatting

Browse Source
This commit is contained in:
mertalev 2024-02-02 23:35:05 -05:00
parent bb56bd3297
commit 073965c47e
No known key found for this signature in database
GPG Key ID: 9181CD92C0A1C5E3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -12,6 +12,7 @@ from huggingface_hub import snapshot_download
from onnx.shape_inference import infer_shapes_path from onnx.shape_inference import infer_shapes_path
from onnx.tools.update_model_dims import update_inputs_outputs_dims from onnx.tools.update_model_dims import update_inputs_outputs_dims
from typing_extensions import Buffer from typing_extensions import Buffer
import ann.ann import ann.ann
from app.models.constants import STATIC_INPUT_PROVIDERS, SUPPORTED_PROVIDERS from app.models.constants import STATIC_INPUT_PROVIDERS, SUPPORTED_PROVIDERS
@ -145,7 +146,7 @@ class InferenceModel(ABC):
# check_model gets called in update_inputs_outputs_dims # check_model gets called in update_inputs_outputs_dims
check_model = onnx.checker.check_model check_model = onnx.checker.check_model
try: try:
def check_model_stub(*args: Any, **kwargs: Any) -> None: def check_model_stub(*args: Any, **kwargs: Any) -> None:
pass pass