bugfix in finetuning script, added gitignore
Browse files- .gitignore +1 -0
- 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",
|