open_flamingo

What's Changed

• Evaluation updates: add RICES + prompt ensembling by @i-gao in https://github.com/mlfoundations/open_flamingo/pull/220
• Pass **kwargs in "generate" function instead of passing each argument by name by @DavidMChan in https://github.com/mlfoundations/open_flamingo/pull/236
• update setup and requirements by @loftusa in https://github.com/mlfoundations/open_flamingo/pull/241

New Contributors

• @ElegantLin made their first contribution in https://github.com/mlfoundations/open_flamingo/pull/229
• @DavidMChan made their first contribution in https://github.com/mlfoundations/open_flamingo/pull/236
• @loftusa made their first contribution in https://github.com/mlfoundations/open_flamingo/pull/241

Full Changelog: https://github.com/mlfoundations/open_flamingo/compare/2.0.0...2.0.1

Major codebase updates, along with five new models.

Changes:

• Add gradient checkpointing, FullyShardedDataParallel
• Remove color jitter when training
• Fix cross-attention bug when calling generate()
• Model releases
  • (CLIP ViT-L-14 / MPT-1B)
  • (CLIP ViT-L-14 / MPT-1B Dolly)
  • (CLIP ViT-L-14 / RedPajama-3B)
  • (CLIP ViT-L-14 / RedPajama-3B Instruct)
  • (CLIP ViT-L-14 / MPT-7B)

Thanks to @chris-alexiuk for fixing!