search

OpenGVLab / LAMM

[NeurIPS 2023 Datasets and Benchmarks Track] LAMM: Multi-Modal Large Language Models and Applications as AI Agents
https://openlamm.github.io/
286 stars 15 forks source link

fix save_model func in src/model/agent.py #33

Closed lighten001 closed 1 year ago

lighten001 commented 1 year ago

for issue https://github.com/OpenLAMM/LAMM/issues/32

  1. only save model in Rank 0 process
  2. consolidate weights in Rank 0 before saving trainable parameters