Prajwal782007 commited on
Commit
ffe65ee
·
1 Parent(s): a4bc605

fix: update TRL requirement to support GRPOTrainer

Browse files
Files changed (1) hide show
  1. scripts/gridmind_grpo_colab.ipynb +1 -1
scripts/gridmind_grpo_colab.ipynb CHANGED
@@ -34,7 +34,7 @@
34
  "outputs": [],
35
  "source": [
36
  "# Install dependencies\n",
37
- "!pip install trl==0.8.6 transformers>=4.41.0 torch accelerate datasets requests -q\n",
38
  "\n",
39
  "import torch\n",
40
  "import sys\n",
 
34
  "outputs": [],
35
  "source": [
36
  "# Install dependencies\n",
37
+ "!pip install trl>=0.14.0 transformers>=4.41.0 torch accelerate datasets requests -q\n",
38
  "\n",
39
  "import torch\n",
40
  "import sys\n",