diff --git a/README.md b/README.md index 92dd76fcb..be993a2b9 100644 --- a/README.md +++ b/README.md @@ -65,7 +65,7 @@ $ gh release create v0.0.0 assets/wheels/vllm-*.whl --title "Forge Wheels v0.0.0 To run SFT on a Llama3 8B model, run ```bash -python -m apps.sft_v2.main --config apps/sft_v2/llama3_8b.yaml +python -m apps.sft.main --config apps/sft/llama3_8b.yaml ``` ### Citation diff --git a/apps/sft_v2/llama3_8b.yaml b/apps/sft/llama3_8b.yaml similarity index 94% rename from apps/sft_v2/llama3_8b.yaml rename to apps/sft/llama3_8b.yaml index 273d2d592..43a690c1e 100644 --- a/apps/sft_v2/llama3_8b.yaml +++ b/apps/sft/llama3_8b.yaml @@ -1,4 +1,4 @@ -# >>> python -m apps.sft_v2.main --config apps/sft_v2/llama3_8b.yaml +# >>> python -m apps.sft.main --config apps/sft/llama3_8b.yaml # Config for supervised full finetuning using a Llama3.1 8B Instruct model diff --git a/apps/sft_v2/main.py b/apps/sft/main.py similarity index 99% rename from apps/sft_v2/main.py rename to apps/sft/main.py index 61b27baa3..27a8036d4 100644 --- a/apps/sft_v2/main.py +++ b/apps/sft/main.py @@ -6,7 +6,7 @@ """To run: -python -m apps.sft_v2.main --config apps/sft_v2/llama3_8b.yaml +python -m apps.sft.main --config apps/sft/llama3_8b.yaml """ diff --git a/apps/sft_v2/qwen3_8b.yaml b/apps/sft/qwen3_8b.yaml similarity index 94% rename from apps/sft_v2/qwen3_8b.yaml rename to apps/sft/qwen3_8b.yaml index 1a07bfa15..2ab88bbd3 100644 --- a/apps/sft_v2/qwen3_8b.yaml +++ b/apps/sft/qwen3_8b.yaml @@ -1,4 +1,4 @@ -# >>> python -m apps.sft_v2.main --config apps/sft_v2/qwen3_8b.yaml +# >>> python -m apps.sft.main --config apps/sft/qwen3_8b.yaml # TODO: required by torchtitan