SYMBOL INDEX (18 symbols across 5 files)

FILE: export_state_dict_checkpoint.py
  function permute (line 57) | def permute(w):
  function unpermute (line 65) | def unpermute(w):
  function translate_state_dict_key (line 73) | def translate_state_dict_key(k):  # noqa: C901

FILE: finetune.py
  function train (line 28) | def train(

FILE: generate.py
  function main (line 26) | def main(

FILE: utils/callbacks.py
  class Stream (line 15) | class Stream(transformers.StoppingCriteria):
    method __init__ (line 16) | def __init__(self, callback_func=None):
    method __call__ (line 19) | def __call__(self, input_ids, scores) -> bool:
  class Iteratorize (line 25) | class Iteratorize:
    method __init__ (line 32) | def __init__(self, func, kwargs={}, callback=None):
    method __iter__ (line 61) | def __iter__(self):
    method __next__ (line 64) | def __next__(self):
    method __enter__ (line 71) | def __enter__(self):
    method __exit__ (line 74) | def __exit__(self, exc_type, exc_val, exc_tb):

FILE: utils/prompter.py
  class Prompter (line 10) | class Prompter(object):
    method __init__ (line 13) | def __init__(self, template_name: str = "", verbose: bool = False):
    method generate_prompt (line 28) | def generate_prompt(
    method get_response (line 50) | def get_response(self, output: str) -> str: