SYMBOL INDEX (37 symbols across 7 files)

FILE: chat/config.py
  class ModelArguments (line 26) | class ModelArguments:
  class DataArguments (line 60) | class DataArguments:
  class TrainingArguments (line 112) | class TrainingArguments(transformers.TrainingArguments):

FILE: chat/dialogues.py
  class DialogueTemplate (line 32) | class DialogueTemplate(ModelHubMixin):
    method get_training_prompt (line 45) | def get_training_prompt(self) -> str:
    method get_inference_prompt (line 56) | def get_inference_prompt(self) -> str:
    method get_dialogue (line 68) | def get_dialogue(self):
    method get_special_tokens (line 80) | def get_special_tokens(self) -> List[str]:
    method copy (line 83) | def copy(self):
    method to_dict (line 93) | def to_dict(self) -> Dict[str, Any]:
    method from_dict (line 97) | def from_dict(cls, data):
    method _save_pretrained (line 107) | def _save_pretrained(self, save_directory: Union[str, Path]) -> None:
    method _from_pretrained (line 114) | def _from_pretrained(
  function get_dialogue_template (line 199) | def get_dialogue_template(template: str) -> DialogueTemplate:
  function prepare_dialogue (line 205) | def prepare_dialogue(example, dialogue_template, is_train=True):
  function mask_user_labels (line 232) | def mask_user_labels(tokenizer, dialogue_template, labels):

FILE: chat/generate.py
  function main (line 25) | def main():

FILE: chat/train.py
  function main (line 44) | def main():

FILE: chat/utils.py
  class StarChatArgumentParser (line 25) | class StarChatArgumentParser(HfArgumentParser):
    method parse_yaml_and_args (line 26) | def parse_yaml_and_args(self, yaml_arg: str, other_args: Optional[List...
  function hf_login (line 61) | def hf_login():

FILE: finetune/finetune.py
  class SavePeftModelCallback (line 18) | class SavePeftModelCallback(TrainerCallback):
    method on_save (line 19) | def on_save(
  class LoadBestPeftModelCallback (line 35) | class LoadBestPeftModelCallback(TrainerCallback):
    method on_train_end (line 36) | def on_train_end(
  function get_args (line 51) | def get_args():
  function chars_token_ratio (line 93) | def chars_token_ratio(dataset, tokenizer, input_column_name="prompt", ou...
  function print_trainable_parameters (line 109) | def print_trainable_parameters(model):
  function prepare_sample_text (line 124) | def prepare_sample_text(example, input_column_name="prompt", output_colu...
  class ConstantLengthDataset (line 130) | class ConstantLengthDataset(IterableDataset):
    method __init__ (line 142) | def __init__(
    method __iter__ (line 163) | def __iter__(self):
  function create_datasets (line 194) | def create_datasets(tokenizer, args):
  function run_training (line 237) | def run_training(args, train_data, val_data):
  function main (line 300) | def main(args):

FILE: finetune/merge_peft_adapters.py
  function get_args (line 8) | def get_args():
  function main (line 16) | def main():