AlexHung29629 commited on
Commit
fcbe13e
·
verified ·
1 Parent(s): e4ff99a

Update ultravox_model.py

Browse files
Files changed (1) hide show
  1. ultravox_model.py +2 -2
ultravox_model.py CHANGED
@@ -204,8 +204,8 @@ class UltravoxModel(transformers.LlamaPreTrainedModel):
204
  try:
205
  inputs_embeds[i, start : start + length] = audio[:length]
206
  except:
207
- print(f{audio.shape=}, {start=}, {length=}, {audio.shape=})
208
- print(f{labels=})
209
  exit(1)
210
 
211
  lm_output = self.language_model.forward(
 
204
  try:
205
  inputs_embeds[i, start : start + length] = audio[:length]
206
  except:
207
+ print(f"{audio.shape=}, {start=}, {length=}, {audio.shape=}")
208
+ print(f"{labels=}")
209
  exit(1)
210
 
211
  lm_output = self.language_model.forward(