[bug fix] re-add model (pytorch#2135)

Co-authored-by: Felipe Mello <[email protected]>
felipemello1 · Dec 9, 2024 · f2bd4bc · f2bd4bc
1 parent 384726f
commit f2bd4bc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/recipes/configs/llama3_2/1B_full_single_device.yaml b/recipes/configs/llama3_2/1B_full_single_device.yaml
@@ -19,9 +19,12 @@
 #
 # This config works only for training on single device.
 
-
 output_dir: /tmp/torchtune/llama3_2_1B/full_single_device # /tmp may be deleted by your system. Change it to your preference.
 
+# Model Arguments
+model:
+  _component_: torchtune.models.llama3_2.llama3_2_1b
+
 # Tokenizer
 tokenizer:
   _component_: torchtune.models.llama3.llama3_tokenizer