From 9c5a04f25fbd2d756fc787b2a50fec9519f68165 Mon Sep 17 00:00:00 2001
From: victpork <224617+victpork@users.noreply.github.com>
Date: Thu, 21 Nov 2024 00:06:22 +1300
Subject: [PATCH] Added code to download weights from new host (#15087)

---
 notebooks/YOLO_NAS_Pretrained_Export.ipynb | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/notebooks/YOLO_NAS_Pretrained_Export.ipynb b/notebooks/YOLO_NAS_Pretrained_Export.ipynb
index a3c303c01..e4e2222da 100644
--- a/notebooks/YOLO_NAS_Pretrained_Export.ipynb
+++ b/notebooks/YOLO_NAS_Pretrained_Export.ipynb
@@ -11,6 +11,18 @@
         "! pip install -q super_gradients==3.7.1"
       ]
     },
+    {
+      "cell_type": "code",
+      "source": [
+        "! sed -i 's/sghub.deci.ai/sg-hub-nv.s3.amazonaws.com/' /usr/local/lib/python3.10/dist-packages/super_gradients/training/pretrained_models.py\n",
+        "! sed -i 's/sghub.deci.ai/sg-hub-nv.s3.amazonaws.com/' /usr/local/lib/python3.10/dist-packages/super_gradients/training/utils/checkpoint_utils.py"
+      ],
+      "metadata": {
+        "id": "NiRCt917KKcL"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -72,4 +84,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}
\ No newline at end of file