Upate OpenVINO to PyTorch 2.6 and fix mismatched shapes error on too many resolution changes

2026-09-19 09:14:35 +02:00 · 2025-02-09 01:17:06 +03:00
parent e578afc1f5
commit 1acbabb276
6 changed files with 34 additions and 20 deletions
@@ -42,7 +42,7 @@ def load_model(device, model_path, model_type="dpt_large_384", optimize=True, he
        network input
    """
    if "openvino" in model_type:
-        from openvino.runtime import Core
+        from openvino import Core

    keep_aspect_ratio = not square