Skip to content

Commit

Permalink
undo post_process_vLLm flag
Browse files Browse the repository at this point in the history
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
  • Loading branch information
Ssukriti committed Sep 23, 2024
1 parent 738239e commit 6afdbfe
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 5 deletions.
2 changes: 0 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -665,8 +665,6 @@ The `fms_acceleration.cli` can do more to search for all available configs, plug
## Inference
Currently, we do *not* offer inference support as part of the library, but we provide a standalone script for running inference on tuned models for testing purposes. For a full list of options run `python scripts/run_inference.py --help`. Note that no data formatting / templating is applied at inference time.

If you are trying to run LoRA inference on vLLM, set the `--post_process_vllm` flag to `True`.

### Running a single example
If you want to run a single example through a model, you can pass it with the `--text` flag.

Expand Down
3 changes: 0 additions & 3 deletions tuning/sft_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -523,7 +523,6 @@ def parse_arguments(parser, json_config=None):
) = parser.parse_dict(json_config, allow_extra_keys=True)
peft_method = json_config.get("peft_method")
exp_metadata = json_config.get("exp_metadata")
post_process_vllm = json_config.get("post_process_vllm")
else:
(
model_args,
Expand All @@ -543,7 +542,6 @@ def parse_arguments(parser, json_config=None):

peft_method = additional.peft_method
exp_metadata = additional.exp_metadata
post_process_vllm = additional.post_process_vllm

if peft_method == "lora":
tune_config = lora_config
Expand All @@ -564,7 +562,6 @@ def parse_arguments(parser, json_config=None):
fusedops_kernels_config,
attention_and_distributed_packing_config,
exp_metadata,
post_process_vllm,
)


Expand Down

0 comments on commit 6afdbfe

Please sign in to comment.