start_audio_token = "<soa>"
end_audio_token = "<eoa>"
end_sequence_token = "<eos>"
device = "cuda"
parser = argparse.ArgumentParser(description="Train a model with configuration.")

    "--config", type=str, help="Path to the config.yaml file", required=True
parser.add_argument(
    "--asr", action="store_true", help="asr/tts",
parser.add_argument(
    "--audio_path", type=str, help="Path to audio if asr is used",
