SYMBOL INDEX (36 symbols across 9 files)

FILE: llm_merging/data.py
  class Dataset (line 3) | class Dataset(object):
    method __init__ (line 4) | def __init__(
    method __len__ (line 14) | def __len__(self):
    method __getitem__ (line 17) | def __getitem__(self, idx):

FILE: llm_merging/evaluation.py
  function convert_dict_of_lists_to_list_of_dicts (line 14) | def convert_dict_of_lists_to_list_of_dicts(dict_of_lists: Dict[Any, List...
  function collate_fn (line 27) | def collate_fn(batch_of_datapoints: List[Dict]) -> Dict[Any, List]:
  function evaluate_dataset (line 48) | def evaluate_dataset(
  function evaluate_model (line 90) | def evaluate_model(

FILE: llm_merging/main.py
  function all_merge_handlers (line 10) | def all_merge_handlers():

FILE: llm_merging/merging/FlanT5Avg.py
  class FlanT5Avg (line 7) | class FlanT5Avg(Merges):
    method __init__ (line 8) | def __init__(self, name):
    method merge (line 40) | def merge(

FILE: llm_merging/merging/LlamaAvg.py
  class LlamaAvg (line 6) | class LlamaAvg(Merges):
    method __init__ (line 7) | def __init__(self, name):
    method merge (line 43) | def merge(

FILE: llm_merging/merging/Merges.py
  class Merges (line 18) | class Merges(object):
    method __init__ (line 20) | def __init__(self, name):
    method get_name (line 44) | def get_name(self):
    method get_model_config (line 47) | def get_model_config(self):
    method _load_base_model (line 50) | def _load_base_model(self):
    method _load_tokenizer (line 59) | def _load_tokenizer(self):
    method predict_multiple_choice (line 101) | def predict_multiple_choice(self, batch):
    method generate (line 111) | def generate(self, batch):
    method _load_huggingface_models_and_configs (line 121) | def _load_huggingface_models_and_configs(self):
    method merge (line 156) | def merge(
    method save_model (line 161) | def save_model(self, output_dir):

FILE: llm_merging/merging/TinyLlamaAvg.py
  class TinyLlamaAvg (line 6) | class TinyLlamaAvg(Merges):
    method __init__ (line 7) | def __init__(self, name):
    method merge (line 42) | def merge(

FILE: llm_merging/model/decoder_functions.py
  function predict_multiple_choice (line 4) | def predict_multiple_choice(
  function generate (line 119) | def generate(
  function tokenize_batch (line 150) | def tokenize_batch(input_tokenizer, target_tokenizer, batch, device):

FILE: llm_merging/model/encoder_decoder_functions.py
  function compute_loss (line 5) | def compute_loss(transformer, tokenizer, batch):
  function predict_multiple_choice (line 41) | def predict_multiple_choice(
  function generate (line 106) | def generate(
  function tokenize_batch (line 130) | def tokenize_batch(tokenizer, batch, device):