lmflow.pipeline.sglang_inferencer#

Attributes#

logger

Classes#

SGLangInferencer

Module Contents#

lmflow.pipeline.sglang_inferencer.logger[source]#
class lmflow.pipeline.sglang_inferencer.SGLangInferencer(model_args: lmflow.args.ModelArguments, data_args: lmflow.args.DatasetArguments, inferencer_args: lmflow.args.InferencerArguments)[source]#

Bases: lmflow.pipeline.base_pipeline.BasePipeline

model_args[source]#
data_args[source]#
inferencer_args[source]#
eos_token_id[source]#
sampling_params[source]#
_parse_args_to_sampling_params(inference_args: lmflow.args.InferencerArguments) → dict[source]#
inference(model: lmflow.models.hf_decoder_model.HFDecoderModel, dataset: lmflow.datasets.Dataset, release_gpu: bool = False, inference_args: lmflow.args.InferencerArguments | None = None)[source]#
save_inference_results(outputs: list[list[str]] | list[list[list[int]]], save_file_path: str)[source]#
load_inference_results(results_path: str) → list[list[str]] | list[list[list[int]]][source]#