fix torch 2.6 unpickling

openvinotoolkit · Feb 3, 2025 · 7b4831f · 7b4831f
1 parent e678798
commit 7b4831f
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/notebooks/bark-text-to-audio/bark-text-to-audio.ipynb b/notebooks/bark-text-to-audio/bark-text-to-audio.ipynb
@@ -142,7 +142,7 @@
    "source": [
     "text_use_small = True\n",
     "\n",
-    "text_encoder = load_model(model_type=\"text\", use_gpu=False, use_small=text_use_small, force_reload=False)\n",
+    "text_encoder = load_model(model_type=\"text\", use_gpu=False, use_small=text_use_small, force_reload=False, weights_only=False)\n",
     "\n",
     "text_encoder_model = text_encoder[\"model\"]\n",
     "tokenizer = text_encoder[\"tokenizer\"]"

diff --git a/notebooks/paddle-ocr-webcam/pre_post_processing.py b/notebooks/paddle-ocr-webcam/pre_post_processing.py
@@ -461,7 +461,7 @@ def draw_ocr_box_txt(image, boxes, txts, scores=None, drop_score=0.5, font_path=
                 try:
                     char_size = font.getbox(c)
                     y_idx = -1
-                except AttributeError:
+                except Exception:
                     char_size = font.getsize(c)
                     y_idx = 1
 

diff --git a/notebooks/softvc-voice-conversion/softvc-voice-conversion.ipynb b/notebooks/softvc-voice-conversion/softvc-voice-conversion.ipynb
@@ -108,7 +108,7 @@
     "\n",
     "%pip install -q \"openvino>=2023.2.0\"\n",
     "clone_repo(\"https://github.com/svc-develop-team/so-vits-svc\", revision=\"4.1-Stable\", add_to_sys_path=False)\n",
-    "%pip install -q --extra-index-url https://download.pytorch.org/whl/cpu  tqdm librosa \"torch>=2.1.0\" \"torchaudio>=2.1.0\" faiss-cpu \"gradio>=4.19\" \"numpy>=1.23.5\" praat-parselmouth"
+    "%pip install -q --extra-index-url https://download.pytorch.org/whl/cpu  tqdm librosa \"torch>=2.1.0,<2.6.0\" \"torchaudio>=2.1.0,<2.6.0\" faiss-cpu \"gradio>=4.19\" \"numpy>=1.23.5\" praat-parselmouth"
    ]
   },
   {

diff --git a/notebooks/yolov10-optimization/yolov10-optimization.ipynb b/notebooks/yolov10-optimization/yolov10-optimization.ipynb
@@ -86,7 +86,7 @@
     "%pip install -q \"nncf>=2.11.0\"\n",
     "%pip install -Uq \"openvino>=2024.3.0\"\n",
     "%pip install -q \"git+https://github.com/THU-MIG/yolov10.git\" --extra-index-url https://download.pytorch.org/whl/cpu\n",
-    "%pip install -q \"torch>=2.1\" \"torchvision>=0.16\" tqdm opencv-python \"gradio>=4.19\" \"matplotlib>=3.9\" --extra-index-url https://download.pytorch.org/whl/cpu"
+    "%pip install -q \"torch>=2.1,<2.6\" \"torchvision>=0.16\" tqdm opencv-python \"gradio>=4.19\" \"matplotlib>=3.9\" --extra-index-url https://download.pytorch.org/whl/cpu"
    ]
   },
   {

diff --git a/notebooks/yolov9-optimization/yolov9-optimization.ipynb b/notebooks/yolov9-optimization/yolov9-optimization.ipynb
@@ -63,7 +63,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%pip install -q \"openvino>=2023.3.0\" \"nncf>=2.8.1\" \"opencv-python\" \"matplotlib>=3.4\" \"seaborn\" \"pandas\" \"scikit-learn\" \"torch\" \"torchvision\" \"tqdm\"  --extra-index-url https://download.pytorch.org/whl/cpu"
+    "%pip install -q \"openvino>=2023.3.0\" \"nncf>=2.8.1\" \"opencv-python\" \"matplotlib>=3.4\" \"seaborn\" \"pandas\" \"scikit-learn\" \"torch<2.6.0\" \"torchvision\" \"tqdm\"  --extra-index-url https://download.pytorch.org/whl/cpu"
    ]
   },
   {