from transformers import AutoTokenizer, AutoModelCausalLM, pipeline model_id = "gpt2" tokenizer = AutoTokenizer.from_pretrained(model_id, cache_dir = "/kaggle/working/augmented") pipe = pipeline("text-generation", model = lm_model, tokenizer = tokenizer, max_new_tokens = 512, devic…

Facing Issue in importing pipelines from transformers

Abdelkareem June 27, 2023, 6:37pm 7

make sure to restart your kernel after
pip install transformers==4.28.0

2 Likes