USC-Applied-NLP-Group
/

SQL-Generation

Model card Files Files and versions Metrics Training metrics Community

licesma commited on 10 days ago

Commit

ef32c53

·

1 Parent(s): 9687f25

fix cuda

Files changed (1) hide show

finetune_model.ipynb +1 -1

finetune_model.ipynb CHANGED Viewed

@@ -213,7 +213,7 @@
     "\n",
     "# Load model with quantization\n",
     "#device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
-    "device_name = 'cuda:0' if torch.cuda.is_available() else 'cpu'\n",
     "device = torch.device(device_name)\n",
     "model = AutoModelForCausalLM.from_pretrained(\n",
     "    model_name, \n",

     "\n",
     "# Load model with quantization\n",
     "#device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
+    "device_name = 'cuda' if torch.cuda.is_available() else 'cpu'\n",
     "device = torch.device(device_name)\n",
     "model = AutoModelForCausalLM.from_pretrained(\n",
     "    model_name, \n",