NameError: name 'init_empty_weights' is not defined on Windows 11 #60
Description
Try to run this:
python ./tts/infer_cli.py --input_wav 'assets/English_prompt.wav' --input_text "As his long promised tariff threat turned into reality this week, top human advisers began fielding a wave of calls from business leaders, particularly in the automotive sector, along with lawmakers who were sounding the alarm." --output_dir ./gen --p_w 2.0 --t_w 3.0
I got this
| loaded 'dur_model' from './checkpoints\duration_lm/model_only_last.ckpt'.
| Missing keys: 0, Unexpected keys: 0
| loaded 'dit' from './checkpoints\diffusion_transformer/model_only_last.ckpt'.
| Missing keys: 0, Unexpected keys: 9
| loaded 'model' from './checkpoints\aligner_lm/model_only_last.ckpt'.
| Missing keys: 0, Unexpected keys: 0
Traceback (most recent call last):
File "D:\Repo_AI\bytedanceMegaTTS3\tts\infer_cli.py", line 267, in <module>
infer_ins = MegaTTS3DiTInfer()
File "D:\Repo_AI\bytedanceMegaTTS3\tts\infer_cli.py", line 90, in __init__
self.build_model(self.device)
File "D:\Repo_AI\bytedanceMegaTTS3\tts\infer_cli.py", line 144, in build_model
self.g2p_model = AutoModelForCausalLM.from_pretrained(self.g2p_exp_name).eval().to(device)
File "D:\DEVSDK\anaconda3\envs\megatts3-env\lib\site-packages\transformers\models\auto\auto_factory.py", line 571, in from_pretrained
return model_class.from_pretrained(
File "D:\DEVSDK\anaconda3\envs\megatts3-env\lib\site-packages\transformers\modeling_utils.py", line 279, in _wrapper
return func(*args, **kwargs)
File "D:\DEVSDK\anaconda3\envs\megatts3-env\lib\site-packages\transformers\modeling_utils.py", line 4333, in from_pretrained
model_init_context = cls.get_init_context(is_quantized, _is_ds_init_called)
File "D:\DEVSDK\anaconda3\envs\megatts3-env\lib\site-packages\transformers\modeling_utils.py", line 3736, in get_init_context
init_contexts = [no_init_weights(), init_empty_weights()]
NameError: name 'init_empty_weights' is not defined
already upgraded the transformers did not fix it. Help?