TideDra / VL-RLHF

A RLHF Infrastructure for Vision-Language Models
Apache License 2.0
85 stars 5 forks source link

请问支持对internvl-1-5的微调吗?如果可以的话显存应该预留多少 #3

Open sunzx8 opened 3 months ago

TideDra commented 3 months ago

您好,未来我们会考虑添加新的模型。您也可以参考docs/CustomizedModel.md添加自定义模型,欢迎您的PR!

delian11 commented 2 months ago

您好,未来我们会考虑添加新的模型。您也可以参考docs/CustomizedModel.md添加自定义模型,欢迎您的PR!

请问您预估internvl-v1-5用这个docs/CustomizedModel.md添加自定义模型在2卡A100上能跑吗?会不会爆显存?