TypeError:pipeline() missing 1 required positional argument :'task'
Minjuner-97 opened this issue · 1 comments
Minjuner-97 commented
from transformers import pipeline
import torch
instruct_pipeline = pipeline(model="databricks/dolly-v2-12b", torch_dtype=torch.float16, trust_remote_code=True, device_map="auto")
GPU:1*V100
transformers:4.3.0
torch:1.2.0
cuda:10.0
cudnn:10.0-7.6.3
result:
TypeError:pipeline() missing 1 required positional argument :'task'
How can I resolve it ?
srowen commented
Your version of transformers is quite old. As you see in the model card and repo here, you should use 4.28 or later.