Spaces:
Running
Running
Commit ·
7597057
1
Parent(s): a3862f1
fix: update dependencies in colab notebook for GRPOTrainer
Browse files
scripts/gridmind_grpo_colab.ipynb
CHANGED
|
@@ -33,17 +33,10 @@
|
|
| 33 |
"metadata": {},
|
| 34 |
"outputs": [],
|
| 35 |
"source": [
|
| 36 |
-
"
|
| 37 |
-
"
|
| 38 |
-
"\n",
|
| 39 |
-
"
|
| 40 |
-
"import sys\n",
|
| 41 |
-
"\n",
|
| 42 |
-
"print(f\"PyTorch: {torch.__version__}\")\n",
|
| 43 |
-
"print(f\"CUDA available: {torch.cuda.is_available()}\")\n",
|
| 44 |
-
"if torch.cuda.is_available():\n",
|
| 45 |
-
" print(f\"GPU: {torch.cuda.get_device_name(0)}\")\n",
|
| 46 |
-
" print(f\"VRAM: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB\")"
|
| 47 |
]
|
| 48 |
},
|
| 49 |
{
|
|
|
|
| 33 |
"metadata": {},
|
| 34 |
"outputs": [],
|
| 35 |
"source": [
|
| 36 |
+
"!pip install trl transformers accelerate datasets unsloth requests pandas matplotlib\n",
|
| 37 |
+
"import os\n",
|
| 38 |
+
"os.makedirs('results', exist_ok=True)\n",
|
| 39 |
+
"print(\"\u2714 All dependencies installed\")\n"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
]
|
| 41 |
},
|
| 42 |
{
|