Spaces:

PEFT
/

causal-language-modeling

Running

stevhliu HF Staff

mimihuhu commited on 7 days ago

Commit

94601dc

verified ·

1 Parent(s): 263af0d

Update prompt_tuning_clm.ipynb (#8)

- Update prompt_tuning_clm.ipynb (e03443e5090e0307c2f9c744a6cb21feb3fe5962)

Co-authored-by: zhang <[email protected]>

Files changed (1) hide show

prompt_tuning_clm.ipynb CHANGED Viewed

@@ -16,7 +16,6 @@
     "from torch.utils.data import DataLoader\n",
     "from transformers import default_data_collator, get_linear_schedule_with_warmup\n",
     "from tqdm import tqdm\n",
-    "from datasets import load_dataset\n",
     "\n",
     "device = \"cuda\"\n",
     "model_name_or_path = \"bigscience/bloomz-560m\"\n",

     "from torch.utils.data import DataLoader\n",
     "from transformers import default_data_collator, get_linear_schedule_with_warmup\n",
     "from tqdm import tqdm\n",
     "\n",
     "device = \"cuda\"\n",
     "model_name_or_path = \"bigscience/bloomz-560m\"\n",