stable-ts: Error when using stable_whisper.load_hf_whisper

I receive the following error when trying to use stable_whisper.load_hf_whisper. I haven’t been able to find a workaround yet.


 in main
    model = stable_whisper.load_hf_whisper('large-v2', device="cuda")
  File "C:\Users\GY\.conda\envs\rich\lib\site-packages\stable_whisper\whisper_word_level\hf_whisper.py", line 170, in load_hf_whisper
    return WhisperHF(model_name, device, flash=flash)
  File "C:\Users\GY\.conda\envs\rich\lib\site-packages\stable_whisper\whisper_word_level\hf_whisper.py", line 54, in __init__
    self._pipe = load_hf_pipe(self._model_name, device, flash=flash)
  File "C:\Users\GY\.conda\envs\rich\lib\site-packages\stable_whisper\whisper_word_level\hf_whisper.py", line 39, in load_hf_pipe
    pipe = pipeline(
  File "C:\Users\GY\.conda\envs\rich\lib\site-packages\transformers\pipelines\__init__.py", line 834, in pipeline
    framework, model = infer_framework_load_model(
  File "C:\Users\GY\.conda\envs\rich\lib\site-packages\transformers\pipelines\base.py", line 269, in infer_framework_load_model
    model = model_class.from_pretrained(model, **kwargs)
  File "C:\Users\GY\.conda\envs\rich\lib\site-packages\transformers\models\auto\auto_factory.py", line 565, in from_pretrained
    return model_class.from_pretrained(
  File "C:\Users\GY\.conda\envs\rich\lib\site-packages\transformers\modeling_utils.py", line 3085, in from_pretrained
    model = cls(config, *model_args, **model_kwargs)
TypeError: __init__() got an unexpected keyword argument 'attn_implementation'

About this issue

  • Original URL
  • State: closed
  • Created 5 months ago
  • Comments: 17 (9 by maintainers)

Commits related to this issue

Most upvoted comments

@dgoryeo It can take more parameters in 133f3235c34aced23c95b10a37df3031b6e4cf0f or version 2.15.5. Full list of parameters: https://huggingface.co/docs/transformers/en/main_classes/text_generation.