model_config instead of cfg.model

mosaicml · Aug 16, 2023 · a4439c5 · a4439c5
1 parent 9ec0f69
commit a4439c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -349,7 +349,7 @@ def main(cfg: DictConfig):
     # Build Model
     with init_context:
         print('Initializing model...')
-        model = build_composer_model(cfg.model, tokenizer)
+        model = build_composer_model(model_config, tokenizer)
 
     # Log number of parameters
     n_params = sum(p.numel() for p in model.parameters())