Remove extra transform model

mosaicml · Jul 19, 2024 · cf710f3 · cf710f3
1 parent 04b5c47
commit cf710f3
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/llmfoundry/models/hf/hf_causal_lm.py b/llmfoundry/models/hf/hf_causal_lm.py
@@ -148,8 +148,6 @@ def __init__(
             prepare_for_fsdp=True,
         )
 
-        model = self.transform_model(model)
-
 
         train_metrics, eval_metrics = ComposerHFCausalLM.build_metrics(
             use_train_metrics=use_train_metrics,