update

fixie-ai · Nov 7, 2024 · 80b88cb · 80b88cb
1 parent 46d7543
commit 80b88cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"):