Update model weights file for JAMR v0.2