src/rime/models/rnn.py [21:28]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        load_from_checkpoint=None
    ):
        self._padded_item_list = [None] + get_top_items(item_df, max_item_size).index.tolist()
        self._truncated_input_steps = truncated_input_steps
        self._collate_fn = functools.partial(
            _collate_fn,
            tokenize={k: i for i, k in enumerate(self._padded_item_list)},
            truncated_input_steps=truncated_input_steps)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/rime/models/transformer.py [41:49]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        load_from_checkpoint=None
    ):

        self._padded_item_list = [None] + get_top_items(item_df, max_item_size).index.tolist()
        self._truncated_input_steps = truncated_input_steps
        self._collate_fn = functools.partial(
            _collate_fn,
            tokenize={k: i for i, k in enumerate(self._padded_item_list)},
            truncated_input_steps=truncated_input_steps)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -