Skip to content

Commit 4f38c14

Browse files
Mark ObozovMark Obozov
authored andcommitted
recipe registry
1 parent 7aea0ca commit 4f38c14

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

torchtune/_recipe_registry.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,8 @@ class Recipe:
6060
file_path="phi3/mini_full_low_memory.yaml",
6161
),
6262
Config(
63-
name="phi4/mini_full_low_memory",
64-
file_path="phi4/mini_full_low_memory.yaml",
63+
name="phi4/full_low_memory",
64+
file_path="phi4/full_low_memory.yaml",
6565
),
6666
Config(
6767
name="qwen2/7B_full_single_device",
@@ -122,7 +122,7 @@ class Recipe:
122122
Config(name="gemma2/9B_full", file_path="gemma2/9B_full.yaml"),
123123
Config(name="gemma2/27B_full", file_path="gemma2/27B_full.yaml"),
124124
Config(name="phi3/mini_full", file_path="phi3/mini_full.yaml"),
125-
Config(name="phi4/mini_full", file_path="phi4/mini_full.yaml"),
125+
Config(name="phi4/full", file_path="phi4/full.yaml"),
126126
Config(name="qwen2/7B_full", file_path="qwen2/7B_full.yaml"),
127127
Config(name="qwen2/0.5B_full", file_path="qwen2/0.5B_full.yaml"),
128128
Config(name="qwen2/1.5B_full", file_path="qwen2/1.5B_full.yaml"),
@@ -262,12 +262,12 @@ class Recipe:
262262
file_path="phi3/mini_qlora_single_device.yaml",
263263
),
264264
Config(
265-
name="phi4/mini_lora_single_device",
266-
file_path="phi4/mini_lora_single_device.yaml",
265+
name="phi4/lora_single_device",
266+
file_path="phi4/lora_single_device.yaml",
267267
),
268268
Config(
269-
name="phi4/mini_qlora_single_device",
270-
file_path="phi4/mini_qlora_single_device.yaml",
269+
name="phi4/qlora_single_device",
270+
file_path="phi4/qlora_single_device.yaml",
271271
),
272272
Config(
273273
name="qwen2/7B_lora_single_device",
@@ -399,7 +399,7 @@ class Recipe:
399399
Config(name="gemma2/9B_lora", file_path="gemma2/9B_lora.yaml"),
400400
Config(name="gemma2/27B_lora", file_path="gemma2/27B_lora.yaml"),
401401
Config(name="phi3/mini_lora", file_path="phi3/mini_lora.yaml"),
402-
Config(name="phi4/mini_lora", file_path="phi4/mini_lora.yaml"),
402+
Config(name="phi4/lora", file_path="phi4/lora.yaml"),
403403
Config(name="qwen2/7B_lora", file_path="qwen2/7B_lora.yaml"),
404404
Config(name="qwen2/0.5B_lora", file_path="qwen2/0.5B_lora.yaml"),
405405
Config(name="qwen2/1.5B_lora", file_path="qwen2/1.5B_lora.yaml"),

0 commit comments

Comments
 (0)