baaivision / Emu

Emu Series: Generative Multimodal Models from BAAI
https://baaivision.github.io/emu2/
Apache License 2.0
1.66k stars 86 forks source link

What's the difference between Emu_pretain.pt and pytorch_model.bin? #36

Open likanchuan opened 1 year ago

likanchuan commented 1 year ago

Hi! Thanks for sharing your awesome work! I have a question on the difference between pytorch_model.bin and Emu_pretain.pt. In other words, if I only want to test the image captioning ability of emu, is Emu_pretain.pt enough?

Moreover, would you mind reporting md5 of each checkpoints please? I meet an error when loading checkpoints so I want to check if the checkpoint is broken. Looking forward to your reply and thanks for your great work again!