#138 · 主分类: MLOps与评测
can-i-finetune-this
估算Hugging Face模型是否适合在本地GPU上进行微调。
项目最后更新:07/23/26
GitHub Stars
792
Forks数量
107
贡献者数量
1
许可证
MIT
收录理由
很多人在下载了14GB的模型后,才发现自己的12GB显卡在第一步训练时就内存溢出,这个CLI工具正是为了解决这种痛点。它能在下载前就估算出某个Hugging Face模型是否能在你的显卡上微调,并且把显存预算拆分成静态估算器通常忽略的部分,比如logits/交叉熵链路以及QLoRA下嵌入层的fp32上转换。如果可行,它还能帮你找到合适的批大小、序列长度、LoRA秩和量化方案,并生成一份可直接运行的PEFT+TRL训练脚本。benchmark和calibrate命令则让你在本地机器上对照真实测量结果验证估算,而不是盲目相信猜测。对于在消费级硬件上微调开源大模型的独立开发者和小团队来说,这能把令人沮丧的试错循环变成快速决策。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。
同类热门项目
unsloth
本地UI,用于运行和训练LLM及扩散模型。支持GGUF、MLX、Qwen3.8、Kimi K3、MiniMax-H3、Gemma 4、FLUX等。
LlamaFactory
统一高效微调100+大语言模型与视觉语言模型(ACL 2024)
airflow
Apache Airflow - 一个以编程方式编写、调度和监控工作流的平台
langfuse
🪢 开源AI工程平台:LLM评估、可观测性、指标、提示管理、游乐场、数据集。与OpenTelemetry、LangChain、OpenAI SDK、LiteLLM等集成。🍊YC W23
netron
神经网络、深度学习和机器学习模型的可视化工具