Prajwal782007 commited on
Commit
7597057
·
1 Parent(s): a3862f1

fix: update dependencies in colab notebook for GRPOTrainer

Browse files
Files changed (1) hide show
  1. scripts/gridmind_grpo_colab.ipynb +4 -11
scripts/gridmind_grpo_colab.ipynb CHANGED
@@ -33,17 +33,10 @@
33
  "metadata": {},
34
  "outputs": [],
35
  "source": [
36
- "# Install dependencies\n",
37
- "!pip install -U git+https://github.com/huggingface/trl.git transformers>=4.41.0 torch accelerate datasets requests -q\\n",
38
- "\n",
39
- "import torch\n",
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
  {