Spaces:
Running
Running
Update prompt_tuning_clm.ipynb
#8
by
mimihuhu
- opened
- prompt_tuning_clm.ipynb +0 -1
prompt_tuning_clm.ipynb
CHANGED
|
@@ -16,7 +16,6 @@
|
|
| 16 |
"from torch.utils.data import DataLoader\n",
|
| 17 |
"from transformers import default_data_collator, get_linear_schedule_with_warmup\n",
|
| 18 |
"from tqdm import tqdm\n",
|
| 19 |
-
"from datasets import load_dataset\n",
|
| 20 |
"\n",
|
| 21 |
"device = \"cuda\"\n",
|
| 22 |
"model_name_or_path = \"bigscience/bloomz-560m\"\n",
|
|
|
|
| 16 |
"from torch.utils.data import DataLoader\n",
|
| 17 |
"from transformers import default_data_collator, get_linear_schedule_with_warmup\n",
|
| 18 |
"from tqdm import tqdm\n",
|
|
|
|
| 19 |
"\n",
|
| 20 |
"device = \"cuda\"\n",
|
| 21 |
"model_name_or_path = \"bigscience/bloomz-560m\"\n",
|