fix sft final

stanford-crfm · Nov 12, 2024 · a3a6db3 · a3a6db3
1 parent 61fe87b
commit a3a6db3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/levanter/main/sft.py b/src/levanter/main/sft.py
@@ -80,6 +80,7 @@ def train(config: SFTConfig):
         raise ValueError("Must specify either --initialize_from_hf or --initialize_from")
     else:
         converter = None
+        model_config = config.model
 
     levanter.initialize(config)