    "--text", type=str, help="Text if tts is used",
args = parser.parse_args()
# Load config
with open(args.config, "r") as file:
    config = yaml.safe_load(file)

    text_tokenized = tokenizer(text, return_tensors="pt")
    text_input_tokens = text_tokenized["input_ids"].to(device)
    soa = tokenizer(start_audio_token, return_tensors="pt")["input_ids"][:, -1:].to(device)
    eoa = tokenizer(end_audio_token, return_tensors="pt")["input_ids"][:, -1:].to(device)
    text_tokens = torch.cat([text_input_tokens, soa], dim=1)
