-
Notifications
You must be signed in to change notification settings - Fork 48
Description
transformers下载并更新到最新版本。把huggingface上面关于模型的文件都下载了,也通过 pip install --upgrade torch命令下载最新的Pytorch库。并访问英伟达toolkit仓库下载toolkit12.4.1版本,最后终端运行了
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
还是出现报错
报错信息
Traceback (most recent call last):
File "C:\Users\Administrator\Desktop\Qwen2-Boundless-main\continuous_conversation.py", line 10, in
model = AutoModelForCausalLM.from_pretrained(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\python\Lib\site-packages\transformers\models\auto\auto_factory.py", line 564, in from_pretrained
return model_class.from_pretrained(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\python\Lib\site-packages\transformers\modeling_utils.py", line 3318, in from_pretrained
raise ImportError(
ImportError: Using low_cpu_mem_usage=True or a device_map requires Accelerate: pip install accelerate
