alex-ht
commited on
Commit
·
fc81040
1
Parent(s):
b044816
update
Browse files- ultravox_model.py +0 -2
ultravox_model.py
CHANGED
@@ -201,8 +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 |
-
length = min(length, audio.shape[0])
|
205 |
-
length = min(inputs_embeds.shape[1] - start, length)
|
206 |
inputs_embeds[i, start : start + length] = audio[:length]
|
207 |
|
208 |
lm_output = self.language_model.forward(
|
|
|
201 |
for i, (audio, start, length) in enumerate(
|
202 |
zip(audio_embeds, audio_token_start_idx, audio_token_len)
|
203 |
):
|
|
|
|
|
204 |
inputs_embeds[i, start : start + length] = audio[:length]
|
205 |
|
206 |
lm_output = self.language_model.forward(
|