modosaic.text.generators.base.generator

modosaic.text.generators.base.generator

TextGenerator

TextGenerator()

Bases: ModalityGenerator[str], ABC

Base class for image-caption generators.

Attributes:

Name	Type	Description
device	device	Torch device selected for inference.
dtype	dtype	Torch dtype selected for inference.
model_specs	HFModelSpec	Hugging Face model metadata.

Initialize device and dtype defaults.

Source code in modosaic/text/generators/base/generator.py

def __init__(self):
    """Initialize device and dtype defaults."""
    self.device, self.dtype = DeviceService.get_device_type()

generate abstractmethod

generate(record)

Generate text for one image record.

Source code in modosaic/text/generators/base/generator.py

@override
@abstractmethod
def generate(self, record: ImageRecord) -> str:
    """Generate text for one image record."""
    raise NotImplementedError()