01-ai / Yi

A series of large language models trained from scratch by developers @01-ai
https://01.ai
Apache License 2.0
7.67k stars 476 forks source link

where can I find the training code or script for YI-VL #505

Closed verigle closed 6 months ago

verigle commented 6 months ago

Reminder

Motivation

I want to fine-tune the model of Yi-VL, but I could not find the training code or script

Solution

can I use the training code of llava ?

Alternatives

No response

Anything Else?

No response

Are you willing to submit a PR?

nuoma commented 6 months ago

Hi, look here https://github.com/01-ai/Yi/pull/368 . You can also try to use other frameworks like SWIFT for VL finetunes.