detector_benchmark.generation.gen_loader
========================================

.. py:module:: detector_benchmark.generation.gen_loader


Classes
-------

.. autoapisummary::

   detector_benchmark.generation.gen_loader.GenLoader


Module Contents
---------------

.. py:class:: GenLoader(model_name: str, gen_params: dict, device: str, gen_tokenizer_only: bool = False)

   .. py:attribute:: model_name


   .. py:attribute:: gen_params


   .. py:attribute:: device


   .. py:attribute:: gen_tokenizer_only


   .. py:method:: 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