fix comments

Signed-off-by: binliu <binliu@nvidia.com>

Author: binliunls

hugging_face/finetune_vista3d_for_hugging_face_pipeline.ipynb

@@ -251,7 +251,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [
     {
@@ -312,7 +312,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -325,7 +325,7 @@
     "\n",
     "config = VISTA3DConfig()\n",
     "vista3d_hf_model = VISTA3DModel(config)\n",
-    "vista3d_hf_model.network.load_state_dict(torch.load(fine_tuned_model_path))\n",
+    "vista3d_hf_model.network.load_state_dict(torch.load(fine_tuned_model_path, weights_only=True))\n",
     "# Save model to hugging face format so that it can be loaded by the from_pretrained API\n",
     "hugging_face_model_path = os.path.join(root_dir, \"huggging_face\")\n",
     "os.makedirs(hugging_face_model_path, exist_ok=True)\n",