import argparse
import torch
import torchaudio
import yaml
from transformers import AutoTokenizer, AutoModelForCausalLM

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