Spaces:
Sleeping
Sleeping
Commit ·
ffe65ee
1
Parent(s): a4bc605
fix: update TRL requirement to support GRPOTrainer
Browse files
scripts/gridmind_grpo_colab.ipynb
CHANGED
|
@@ -34,7 +34,7 @@
|
|
| 34 |
"outputs": [],
|
| 35 |
"source": [
|
| 36 |
"# Install dependencies\n",
|
| 37 |
-
"!pip install trl=
|
| 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",
|