From ea82414b534bc75bc85302595b48284df2f3f602 Mon Sep 17 00:00:00 2001 From: Kartikay Khandelwal Date: Wed, 20 Mar 2024 12:49:43 -0700 Subject: [PATCH] fix test --- torchtune/__init__.py | 2 +- torchtune/_cli/ls.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/torchtune/__init__.py b/torchtune/__init__.py index 576ca44cf5..89fd919f43 100644 --- a/torchtune/__init__.py +++ b/torchtune/__init__.py @@ -18,7 +18,7 @@ "full_finetune_distributed.py": ["full_finetune_distributed.yaml"], "lora_finetune_single_device.py": ["lora_finetune_single_device.yaml"], "lora_finetune_distributed.py": ["lora_finetune_distributed.yaml"], - "alpaca_generate.py": ["generate.yaml"], + "alpaca_generate.py": ["alpaca_generate.yaml"], } diff --git a/torchtune/_cli/ls.py b/torchtune/_cli/ls.py index ca5f868adb..ce1369c2b2 100644 --- a/torchtune/_cli/ls.py +++ b/torchtune/_cli/ls.py @@ -46,7 +46,7 @@ def main(): RECIPE CONFIG full_finetune_distributed.py full_finetune_distributed.yaml lora_finetune_distributed.py lora_finetune_distributed.yaml - alpaca_generate.py generate.yaml + alpaca_generate.py alpaca_generate.yaml To run one of these recipes: $ tune full_finetune_single_device --config full_finetune_single_device