我目前在使用提供的自动训练工具在 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
直到昨天,这段代码在 https://github.com/huggingface/autotrain-advanced 用于微调 LLM 的 Colab 笔记本上运行良好,现在出现此错误。
这个问题有什么解决办法吗?