minor

Signed-off-by: sahil suneja <[email protected]>
foundation-model-stack · Sep 10, 2024 · 879ae21 · 879ae21
1 parent cd4c9c3
commit 879ae21
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/speculator/train_speculator_utils.py b/speculator/train_speculator_utils.py
@@ -432,8 +432,6 @@ def train_speculator(
         is_compiled=cfg.use_torch_compile,
     )
 
-    return train_loss
-
 
 class EmbedGPTBigCode(GPTBigCode):
     # Overrides the forward function of GPTBigCode to allow returning embedding vectors