ValueError: Could not load model bigcode/starcoder with any of the following classes

[24mlight]

pip freeze:

accelerate==0.19.0 anyio==3.6.2 boto3==1.26.135 botocore==1.29.135 certifi==2023.5.7 charset-normalizer==3.1.0 click==8.1.3 fastapi==0.95.2 filelock==3.12.0 fsspec==2023.5.0 h11==0.14.0 huggingface-hub==0.14.1 idna==3.4 jmespath==1.0.1 numpy==1.24.3 nvidia-cublas-cu11==11.10.3.66 nvidia-cuda-nvrtc-cu11==11.7.99 nvidia-cuda-runtime-cu11==11.7.99 nvidia-cudnn-cu11==8.5.0.96 packaging==23.1 psutil==5.9.5 pydantic==1.10.7 python-dateutil==2.8.2 pytorch-pretrained-bert==0.6.2 PyYAML==6.0 regex==2023.5.5 requests==2.30.0 s3transfer==0.6.1 six==1.16.0 sniffio==1.3.0 starlette==0.27.0 tokenizers==0.13.3 torch==1.13.1 tqdm==4.65.0 transformers==4.28.1 typing_extensions==4.5.0 urllib3==1.26.15 uvicorn==0.22.0 `

python main.py --host=0.0.0.0 --port=7088 --pretrained="bigcode/starcoder" Traceback (most recent call last): File "/data/sd-webui/7099/main.py", line 38, in main() File "/data/sd-webui/7099/main.py", line 33, in main generator = StarCoder(args.pretrained, device_map='auto') File "/data/sd-webui/7099/generators.py", line 17, in init self.pipe: Pipeline = pipeline( File "/data/sd-webui/7099/venv/lib/python3.10/site-packages/transformers/pipelines/init.py", line 779, in pipeline framework, model = infer_framework_load_model( File "/data/sd-webui/7099/venv/lib/python3.10/site-packages/transformers/pipelines/base.py", line 271, in infer_framework_load_model raise ValueError(f"Could not load model {model} with any of the following classes: {class_tuple}.") ValueError: Could not load model bigcode/starcoder with any of the following classes: (<class 'transformers.models.auto.modeling_auto.AutoModelForCausalLM'>, <class 'transformers.models.gpt_bigcode.modeling_gpt_bigcode.GPTBigCodeForCausalLM'>).