alex-ht
commited on
Commit
•
09c54e5
1
Parent(s):
6a4a7d5
update
Browse files- ultravox_model.py +0 -1
ultravox_model.py
CHANGED
@@ -201,7 +201,6 @@ class UltravoxModel(transformers.LlamaPreTrainedModel):
|
|
201 |
for i, (audio, start, length) in enumerate(
|
202 |
zip(audio_embeds, audio_token_start_idx, audio_token_len)
|
203 |
):
|
204 |
-
print(f"{length=}, {audio.shape=}, {start=}, {audio.shape=}, {inputs_embeds.shape=}")
|
205 |
length = min(length, audio.shape[0])
|
206 |
inputs_embeds[i, start : start + length] = audio[:length]
|
207 |
|
|
|
201 |
for i, (audio, start, length) in enumerate(
|
202 |
zip(audio_embeds, audio_token_start_idx, audio_token_len)
|
203 |
):
|
|
|
204 |
length = min(length, audio.shape[0])
|
205 |
inputs_embeds[i, start : start + length] = audio[:length]
|
206 |
|