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
- replace `None` timestamp in `WhipserHF.transcribe()` -added logic to replace `None` timestamps returned by Hugging Face Whisper models with adjacent timestamps to prevent cases such as https://github... — committed to jianfch/stable-ts by jianfch 5 months ago
- changed `load_hf_pipe()` -changed `.whisper_word_level.hf_whisper.load_hf_pipe()` load model via a different method to prevent (https://github.com/jianfch/stable-ts/issues/306#issuecomment-1931637245... — committed to jianfch/stable-ts by jianfch 5 months ago
@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.