* fix untrained reserved tokens * save model after fixing untrained embeddings * don't need fsdp conditional here