fix

mosaicml · Mar 29, 2024 · 133d6ba · 133d6ba
1 parent f32bb33
commit 133d6ba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/a_scripts/inference/test_convert_composer_to_hf.py b/tests/a_scripts/inference/test_convert_composer_to_hf.py
@@ -725,7 +725,8 @@ def test_huggingface_conversion_callback(
         device_batch_size,
     )
 
-    original_model = build_composer_model(model, model_cfg, tokenizer)
+    original_model = build_composer_model(model_cfg['name'], model_cfg,
+                                          tokenizer)
     optimizer_name = optimizer_config.pop('name')
     optimizer = build_optimizer(original_model, optimizer_name,
                                 optimizer_config)