From 7d52b2551612cfac4d4fcb16ff819e8b703169e4 Mon Sep 17 00:00:00 2001 From: Felipe Mello Date: Mon, 9 Dec 2024 10:42:51 -0800 Subject: [PATCH] add model --- recipes/configs/llama3_2/1B_full_single_device.yaml | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/recipes/configs/llama3_2/1B_full_single_device.yaml b/recipes/configs/llama3_2/1B_full_single_device.yaml index ddeab81957..e24fc56219 100644 --- 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