Method forward

models/mesh_xl/tokenizer.py:137–145

(self, data_dict: dict)

    
    
    def forward(self, data_dict: dict) -> dict:
        
        encoder_output = self.tokenize(data_dict)
        decoder_output = self.detokenize(
            input_ids = encoder_output['codes'], 
        )
        data_dict.update(encoder_output)
        data_dict.update(decoder_output)
        return data_dict