Fix blbooksgenre notebook failures with pytorch update

Azure · Oct 3, 2024 · 23a2a61 · 23a2a61
1 parent 5e5fc2c
commit 23a2a61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb b/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb
@@ -88,7 +88,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "rai_example_version_string = \"38\""
+    "rai_example_version_string = \"39\""
    ]
   },
   {
@@ -508,7 +508,7 @@
     "        python_model=my_mlflow,\n",
     "        registered_model_name=registered_name,\n",
     "        artifact_path=registered_name,\n",
-    "        pip_requirements=['mlflow', 'torch~=2.0.1', 'transformers>=4.17.0,<=4.44.0'])\n",
+    "        pip_requirements=['mlflow', 'torch>=2.2.2', 'transformers>=4.17.0,<=4.44.0'])\n",
     "\n",
     "    _logger.info(\"Writing JSON\")\n",
     "    dict = {\"id\": \"{0}:1\".format(registered_name)}\n",