Spaces:

Prajwal782007
/

Gridmind

Sleeping

Prajwal782007 commited on Apr 25

Commit

c77108e

1 Parent(s): e9f731a

fix: add missing torch import in Step 4 cell

Files changed (1) hide show

scripts/gridmind_grpo_colab.ipynb CHANGED Viewed

@@ -326,6 +326,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
     "from transformers import AutoTokenizer, AutoModelForCausalLM\n",
     "\n",
     "MODEL_NAME = \"Qwen/Qwen2.5-1.5B-Instruct\"\n",

    "metadata": {},
    "outputs": [],
    "source": [
+    "import torch\n",
     "from transformers import AutoTokenizer, AutoModelForCausalLM\n",
     "\n",
     "MODEL_NAME = \"Qwen/Qwen2.5-1.5B-Instruct\"\n",