From 80b88cb57e59c26b9cf726d5a2c344371551ab12 Mon Sep 17 00:00:00 2001 From: Zhongqiang Huang Date: Wed, 6 Nov 2024 16:13:04 -0800 Subject: [PATCH] update --- ultravox/training/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ultravox/training/train.py b/ultravox/training/train.py index 52be715..5281d80 100644 --- a/ultravox/training/train.py +++ b/ultravox/training/train.py @@ -139,8 +139,8 @@ def train(args: config_base.TrainConfig): model = ultravox_model.UltravoxModel(config) assert model.get_input_embeddings().num_embeddings == len( - text_tokenizer - ), f"Model and tokenizer mismatch: {model.get_input_embeddings().num_embeddings} != {len(text_tokenizer)}" + text_tokenizer + ), f"Model and tokenizer mismatch: {model.get_input_embeddings().num_embeddings} != {len(text_tokenizer)}" model.language_model.config.use_cache = False if args.disable_layerdrop and hasattr(model.audio_tower.config, "layerdrop"):