farzadab commited on
Commit
87ab7ad
·
verified ·
1 Parent(s): d160684

Update ultravox_pipeline.py

Browse files
Files changed (1) hide show
  1. ultravox_pipeline.py +2 -3
ultravox_pipeline.py CHANGED
@@ -33,15 +33,14 @@ class UltravoxPipeline(transformers.Pipeline):
33
  model.config.audio_model_id or model.config.audio_config._name_or_path
34
  )
35
 
 
 
36
  self.processor = UltravoxProcessor(
37
  audio_processor=audio_processor,
38
  tokenizer=tokenizer,
39
  stack_factor=model.config.stack_factor,
40
- audio_context_size=model.audio_tower_context_length,
41
  )
42
 
43
- super().__init__(model=model, tokenizer=tokenizer, **kwargs)
44
-
45
  def _sanitize_parameters(self, **kwargs):
46
  generation_keys = ["temperature", "max_new_tokens", "repetition_penalty"]
47
  generation_kwargs = {k: kwargs[k] for k in kwargs if k in generation_keys}
 
33
  model.config.audio_model_id or model.config.audio_config._name_or_path
34
  )
35
 
36
+ super().__init__(model=model, tokenizer=tokenizer, **kwargs)
37
+
38
  self.processor = UltravoxProcessor(
39
  audio_processor=audio_processor,
40
  tokenizer=tokenizer,
41
  stack_factor=model.config.stack_factor,
 
42
  )
43
 
 
 
44
  def _sanitize_parameters(self, **kwargs):
45
  generation_keys = ["temperature", "max_new_tokens", "repetition_penalty"]
46
  generation_kwargs = {k: kwargs[k] for k in kwargs if k in generation_keys}