detector_benchmark.generation.gen_loader¶

Classes¶

Module Contents¶

class detector_benchmark.generation.gen_loader.GenLoader(model_name: str, gen_params: dict, device: str, gen_tokenizer_only: bool = False)¶

model_name¶

gen_params¶

device¶

gen_tokenizer_only = False¶

load() → tuple[torch.nn.Module, detector_benchmark.generation.generator.LLMGenerator, detector_benchmark.utils.configs.ModelConfig]¶: Load the specifed generator model (from init) and tokenizer

Returns:¶

torch.nn.Module: The loaded generator model LLMGenerator: The loaded generator model ModelConfig: The configuration of the generator model