init_empty_weights is not defined.

[HeartDawnWY]

python: 3.12.4
windows10
源码安装

执行 python examples/web/webui.py 报错

[INFO] all download tasks finished.
[+0900 20250406 09:42:11] [INFO] ChatTTS | dl | checking assets...
[+0900 20250406 09:42:14] [INFO] ChatTTS | dl | all assets are already latest.
[+0900 20250406 09:42:16] [WARN] ChatTTS | gpu | no GPU or NPU found, use CPU instead
[+0900 20250406 09:42:16] [INFO] ChatTTS | core | use device cpu
[+0900 20250406 09:42:18] [INFO] ChatTTS | core | vocos loaded.
[+0900 20250406 09:42:19] [INFO] ChatTTS | core | dvae loaded.
[+0900 20250406 09:42:19] [INFO] ChatTTS | core | embed loaded.
Traceback (most recent call last):
File "H:\python\ChatTTS\examples\web\webui.py", line 287, in
main()
File "H:\python\ChatTTS\examples\web\webui.py", line 268, in main
if load_chat(args.custom_path, args.coef):
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "H:\python\ChatTTS\examples\web\funcs.py", line 67, in load_chat
ret = chat.load(coef=coef)
^^^^^^^^^^^^^^^^^^^^
File "H:\python\ChatTTS\ChatTTS\core.py", line 151, in load
return self._load(
^^^^^^^^^^^
File "e:\env\chattts\Lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "H:\python\ChatTTS\ChatTTS\core.py", line 354, in _load
gpt.load_pretrained(gpt_ckpt_path, embed_path, experimental=experimental)
File "H:\python\ChatTTS\ChatTTS\model\gpt.py", line 75, in load_pretrained
self.gpt: LlamaModel = LlamaModel.from_pretrained(gpt_folder).to(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "e:\env\chattts\Lib\site-packages\transformers\modeling_utils.py", line 279, in _wrapper
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "e:\env\chattts\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 "e:\env\chattts\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

[zbowling]

huggingface/transformers#37326

[mkyuangithub]

pip install accelerate
then solve your issue

[HeartDawnWY]

pip install accelerate then solve your issue

Thank you, I can run it now.