From a3a6db35628fde6ab860dd424274da7af8adce52 Mon Sep 17 00:00:00 2001
From: Ahmed Ahmed <ahmedah@stanford.edu>
Date: Mon, 11 Nov 2024 17:08:26 -0800
Subject: [PATCH] fix sft final

---
 src/levanter/main/sft.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/levanter/main/sft.py b/src/levanter/main/sft.py
index c3dc60eab..71e3c3ef7 100644
--- 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)