Update requirements

ai-forever · Dec 9, 2022 · 84b20a8 · 84b20a8
1 parent 11cb925
commit 84b20a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/ruGPT3XL_finetune_example.ipynb b/examples/ruGPT3XL_finetune_example.ipynb
@@ -5,7 +5,7 @@
    "metadata": {},
    "source": [
     "# Finetuning ruGPT3Xl example\n",
-    "We test this pipeline on `torch==1.12.1+cu113`, `A100 GPU` and it takes around `32GB GPU memory`."
+    "We test this pipeline on `torch 1.7.1+cu110, cuda 11.0, triton 1.0.0, deepspeed 0.7.4`, `A100 GPU` and it takes around `32GB GPU memory`."
    ]
   },
   {