SYMBOL INDEX (33 symbols across 6 files)

FILE: src/configuration_model.py
  class ImplicitModelConfig (line 3) | class ImplicitModelConfig(PretrainedConfig):
    method __init__ (line 4) | def __init__(

FILE: src/data.py
  function extract_answer (line 8) | def extract_answer(text):
  function extract_cot (line 18) | def extract_cot(text):
  class CoTDataset (line 28) | class CoTDataset(Dataset):
    method __init__ (line 29) | def __init__(self, tokenizer, file_path, max_length=-1, max_size=-1):
    method __len__ (line 75) | def __len__(self):
    method __getitem__ (line 79) | def __getitem__(self, i):
  class CoTDataCollator (line 89) | class CoTDataCollator:
    method __init__ (line 95) | def __init__(self, tokenizer):
    method __call__ (line 98) | def __call__(self, examples):
    method _tensorize_batch (line 116) | def _tensorize_batch(self, examples):

FILE: src/generate.py
  function evaluate (line 24) | def evaluate(dataloader, tokenizer, device, ctx, model, max_new_tokens):
  function main (line 67) | def main():

FILE: src/model.py
  class ImplicitModel (line 13) | class ImplicitModel(nn.Module):
    method __init__ (line 14) | def __init__(self, config, reinitialize_weights=False):
    method forward (line 23) | def forward(self, input_ids, position_ids=None, output_attentions=False):
    method compute_loss (line 30) | def compute_loss(self, input_ids, labels, position_ids=None, output_at...
    method generate (line 52) | def generate(self, input_ids, max_new_tokens=512, num_beams=1, stop_on...
    method from_pretrained (line 118) | def from_pretrained(self, pretrained_path):
    method save_pretrained (line 125) | def save_pretrained(self, save_directory):

FILE: src/train.py
  function compute_lambda_distribution (line 25) | def compute_lambda_distribution(removal_smoothing_lambda, truncate_lengt...
  function evaluate (line 39) | def evaluate(dataloader, tokenizer, device, ctx, model, max_new_tokens, ...
  function main (line 130) | def main():

FILE: src/utils.py
  function save_model (line 5) | def save_model(model, tokenizer, model_dir):
  function batch_ids (line 12) | def batch_ids(input_ids_list, pad_token_id, device, dtype):
  function get_sep_position (line 22) | def get_sep_position(input_ids, sep_id, skip=0):
  class DoubleEOSStoppingCriteria (line 36) | class DoubleEOSStoppingCriteria(StoppingCriteria):
    method __init__ (line 37) | def __init__(self, eos_token_id):
    method __call__ (line 42) | def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTen...
  class DoubleEOSLogitsProcessor (line 50) | class DoubleEOSLogitsProcessor(LogitsProcessor):
    method __init__ (line 51) | def __init__(self, eos_token_id):
    method __call__ (line 56) | def __call__(self, input_ids, scores):