        {"additional_special_tokens": [start_audio_token, end_audio_token]}
    )
    n_tokens = len(tokenizer)
    print("Not audio tokens:", n_tokens)
    start_audio_token_id = tokenizer(start_audio_token)["input_ids"][-1]

    tokens_config = get_start_tokens(config["quantizer"], n_tokens)
    quantizer = AudioTokenizer(config["quantizer"], tokens_config)
    codebook_size = config["quantizer"]["speech"]["n_new_tokens"] + config["quantizer"]["wav"]["n_new_tokens"]
    train_dataset, val_dataset = load_data(data, tokenizer, quantizer, config)
    model.resize_token_embeddings(n_tokens + codebook_size)
