Update ultravox_pipeline.py
Browse files- ultravox_pipeline.py +2 -2
 
    	
        ultravox_pipeline.py
    CHANGED
    
    | 
         @@ -33,14 +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 | 
         
             
                    )
         
     | 
| 41 | 
         | 
| 42 | 
         
            -
                    super().__init__(model=model, tokenizer=tokenizer, **kwargs)
         
     | 
| 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}
         
     | 
| 
         | 
|
| 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}
         
     |