DeanGumas commited on
Commit
15c124e
·
1 Parent(s): 837f6f7

bugfix in finetuning script, added gitignore

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. finetune_model.ipynb +1 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ *.pyc
finetune_model.ipynb CHANGED
@@ -145,7 +145,7 @@
145
  " )\n",
146
  "\n",
147
  "# Load model with quantization\n",
148
- "device_name = 'cuda' if torch.cuda.is_available() else 'cpu'\n",
149
  "device = torch.device(device_name)\n",
150
  "model = AutoModelForCausalLM.from_pretrained(\n",
151
  " model_name, \n",
 
145
  " )\n",
146
  "\n",
147
  "# Load model with quantization\n",
148
+ "device_name = 'cuda:0' if torch.cuda.is_available() else 'cpu'\n",
149
  "device = torch.device(device_name)\n",
150
  "model = AutoModelForCausalLM.from_pretrained(\n",
151
  " model_name, \n",