From 84b20a81ddef985cb5cd833b3e1443e7edf56b4f Mon Sep 17 00:00:00 2001 From: Anton Emelyanov Date: Fri, 9 Dec 2022 13:51:03 +0300 Subject: [PATCH] Update requirements --- examples/ruGPT3XL_finetune_example.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/ruGPT3XL_finetune_example.ipynb b/examples/ruGPT3XL_finetune_example.ipynb index f190f6a..627748c 100644 --- 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`." ] }, {