Dumb mistake, added to init

pytorch · Mar 22, 2024 · 004a877 · 004a877
1 parent a5e4fd4
commit 004a877
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchtune/__init__.py b/torchtune/__init__.py
@@ -20,7 +20,7 @@
     "lora_finetune_single_device.py": ["lora_finetune_single_device.yaml"],
     "lora_finetune_distributed.py": ["lora_finetune_distributed.yaml"],
     "alpaca_generate.py": ["alpaca_generate.yaml"],
-    "eleuther_eval.py": ["llama2_eleuther_eval.yaml"],
+    "eleuther_eval.py": ["eleuther_eval.yaml"],
 }