Add mistral to registry

pytorch · Apr 1, 2024 · 72c9b77 · 72c9b77
1 parent 21170f9
commit 72c9b77
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/torchtune/_recipe_registry.py b/torchtune/_recipe_registry.py
@@ -35,6 +35,10 @@ class Recipe:
                 name="llama2/7B_full_single_device_low_memory",
                 file_path="llama2/7B_full_single_device_low_memory.yaml",
             ),
+            Config(
+                name="mistral/7B_full",
+                file_path="mistral/7B_full.yaml",
+            ),
         ],
         supports_distributed=False,
     ),
@@ -44,6 +48,7 @@ class Recipe:
         configs=[
             Config(name="llama2/7B_full", file_path="llama2/7B_full.yaml"),
             Config(name="llama2/13B_full", file_path="llama2/13B_full.yaml"),
+            Config(name="mistral/7B_full", file_path="mistral/7B_full.yaml"),
         ],
         supports_distributed=True,
     ),
@@ -59,6 +64,10 @@ class Recipe:
                 name="llama2/7B_qlora_single_device",
                 file_path="llama2/7B_qlora_single_device.yaml",
             ),
+            Config(
+                name="mistral/7B_lora",
+                file_path="mistral/7B_lora.yaml",
+            ),
         ],
         supports_distributed=False,
     ),
@@ -68,6 +77,7 @@ class Recipe:
         configs=[
             Config(name="llama2/7B_lora", file_path="llama2/7B_lora.yaml"),
             Config(name="llama2/13B_lora", file_path="llama2/13B_lora.yaml"),
+            Config(name="mistral/7B_lora", file_path="mistral/7B_lora.yaml"),
         ],
         supports_distributed=True,
     ),