From d8cc95806256f730c01d25cfc3180108e3e7d184 Mon Sep 17 00:00:00 2001
From: HassanJbara <hasan-account2012-3@live.com>
Date: Thu, 16 Oct 2025 17:00:46 +0200
Subject: [PATCH] Remove no op model compile

---
 flame/models/parallelize_fla.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/flame/models/parallelize_fla.py b/flame/models/parallelize_fla.py
index 37178af1..141fe59c 100644
--- a/flame/models/parallelize_fla.py
+++ b/flame/models/parallelize_fla.py
@@ -399,9 +399,6 @@ def apply_compile(model: nn.Module):
         lm_head = torch.compile(getattr(model, lm_head_key), fullgraph=True)
         model.register_module(lm_head_key, lm_head)
 
-    logger.info("Compiling the entire model with torch.compile")
-    model = torch.compile(model)
-
 
 def apply_fsdp(
     model: nn.Module,