Update torchtune/models/phi4/_model_builders.py

Co-authored-by: Felipe Mello <[email protected]>
pytorch · Jan 14, 2025 · bdf478f · bdf478f
1 parent 1d03294
commit bdf478f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/torchtune/models/phi4/_model_builders.py b/torchtune/models/phi4/_model_builders.py
@@ -75,9 +75,7 @@ def lora_phi4_mini(
     """
     Builder for creating a Phi4 (14b) model with LoRA enabled.
 
-    The Phi4 defaults are the same as in :func:`~torchtune.models.phi4.phi4_mini`,
-    while LoRA default params are based on
-    https://github.com/tloen/alpaca-lora/blob/8bb8579e403dc78e37fe81ffbb253c413007323f/finetune.py#L41-L43.
+    The Phi4 defaults are the same as in :func:`~torchtune.models.phi4.phi4_mini`.
 
     Args:
         lora_attn_modules (List[LORA_ATTN_MODULES]): list of which linear layers