Method __init__

pre_encode.py:40–55

(
        self,
        model,
        output_path,
        is_discrete=False,
        model_half=False,
        model_config=None,
        dataset_config=None,
        sample_size=1920000,
        no_pad=False,
        args_dict=None
    )

class PreEncodedLatentsInferenceWrapper(pl.LightningModule):
    def __init__(
        self,
        model,
        output_path,
        is_discrete=False,
        model_half=False,
        model_config=None,
        dataset_config=None,
        sample_size=1920000,
        no_pad=False,
        args_dict=None
    ):
        super().__init__()
        self.save_hyperparameters(ignore=['model'])
        self.model = model
        self.output_path = Path(output_path)

    def prepare_data(self):
        # runs on rank 0