change directory name

vllm-project · Feb 10, 2025 · a55a427 · a55a427
1 parent 7f49448
commit a55a427
Show file tree

Hide file tree

Showing 15 changed files with 9 additions and 405 deletions.
diff --git a/examples/trl_mixin/ex_trl_constant.py b/examples/trl_mixin/ex_trl_constant.py
@@ -3,7 +3,7 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from trl import DataCollatorForCompletionOnlyLM
 
-from llmcompressor.arg_parser import TrainingArguments
+from llmcompressor.args import TrainingArguments
 
 model_path = "neuralmagic/Llama-2-7b-pruned50-retrained"
 output_dir = "./output_trl_sft_test_7b_gsm8k_sft_data"

diff --git a/examples/trl_mixin/ex_trl_distillation.py b/examples/trl_mixin/ex_trl_distillation.py
@@ -1,7 +1,7 @@
 from sft_trainer import SFTTrainer
 from transformers import AutoModelForCausalLM, AutoTokenizer, DefaultDataCollator
 
-from llmcompressor.arg_parser import DatasetArguments, TrainingArguments
+from llmcompressor.args import DatasetArguments, TrainingArguments
 from llmcompressor.transformers import TextGenerationDataset
 
 model_path = "neuralmagic/Llama-2-7b-pruned50-retrained"

diff --git a/examples/trl_mixin/sft_trainer.py b/examples/trl_mixin/sft_trainer.py
@@ -1,7 +1,7 @@
 from trl import SFTConfig as TRLSFTConfig
 from trl import SFTTrainer as TRLSFTTrainer
 
-from llmcompressor.arg_parser import TrainingArguments
+from llmcompressor.args import TrainingArguments
 from llmcompressor.transformers.finetune.session_mixin import SessionManagerMixIn
 
 __all__ = ["SFTTrainer"]

diff --git a/src/llmcompressor/arg_parser/README.md b/src/llmcompressor/arg_parser/README.md
diff --git a/src/llmcompressor/arg_parser/__init__.py b/src/llmcompressor/arg_parser/__init__.py
diff --git a/src/llmcompressor/arg_parser/dataset_arguments.py b/src/llmcompressor/arg_parser/dataset_arguments.py
diff --git a/src/llmcompressor/arg_parser/model_arguments.py b/src/llmcompressor/arg_parser/model_arguments.py
diff --git a/src/llmcompressor/arg_parser/recipe_arguments.py b/src/llmcompressor/arg_parser/recipe_arguments.py