AutoTrain 高级 CLI:错误:无法识别的参数:--fp16 --use-int4

问题描述 投票:0回答:1

我目前在使用提供的自动训练工具在 Colab 笔记本中使用 LLM 模型微调数据时遇到问题。错误消息表明 autotrain 无法识别参数“--fp16”和“--use-int4”。我已经检查了文档和语法,但问题仍然存在。您能否提供解决此问题的指导或提供有关任何潜在解决方案的见解?谢谢你。

/usr/local/lib/python3.10/dist-packages/torchvision/io/image.py:13:
 UserWarning: Failed to load image Python extension: '/usr/local/lib/python3.10/dist-packages/torchvision/image.so: undefined symbol: _ZN3c104cuda9SetDeviceEi'If you don't plan on using image functionality from`torchvision.io`, you can ignore this warning. Otherwise, there might be something wrong with your environment. Did you have `libjpeg`or`libpng`installed before building`torchvision` from source?   warn( usage: autotrain <command> [<args>] AutoTrain advanced CLI: error: unrecognized arguments: --fp16 --use-int4

screenshot of the error

直到昨天,这段代码在 https://github.com/huggingface/autotrain-advanced 用于微调 LLM 的 Colab 笔记本上运行良好,现在出现此错误。

machine-learning chatbot large-language-model fine-tuning peft
1个回答
0
投票

这个问题有什么解决办法吗?

© www.soinside.com 2019 - 2024. All rights reserved.