AttributeError: module 'torch.library' has no attribute 'register_fake'

Python应用报错如下：

Traceback (most recent call last):
File “/root/miniconda3/lib/python3.10/site-packages/streamlit/runtime/scriptrunner/exec_code.py”, line 88, in exec_func_with_error_handling
result = func()
File “/root/miniconda3/lib/python3.10/site-packages/streamlit/runtime/scriptrunner/script_runner.py”, line 590, in code_to_exec
exec(code, module.dict)
File “/root/VideoLingo/st.py”, line 3, in
from core.st_utils.imports_and_utils import *
File “/root/VideoLingo/core/init.py”, line 3, in
from . import (
File “/root/VideoLingo/core/_10_gen_audio.py”, line 16, in
from core.tts_backend.tts_main import tts_main
File “/root/VideoLingo/core/tts_backend/tts_main.py”, line 16, in
from core.tts_backend.index_tts import index_tts
File “/root/VideoLingo/core/tts_backend/index_tts.py”, line 3, in
from indextts.infer_v2 import IndexTTS2
File “/root/miniconda3/lib/python3.10/site-packages/indextts/infer_v2.py”, line 20, in
from indextts.gpt.model_v2 import UnifiedVoice
File “/root/miniconda3/lib/python3.10/site-packages/indextts/gpt/model_v2.py”, line 9, in
from indextts.gpt.transformers_gpt2 import GPT2PreTrainedModel, GPT2Model
File “/root/miniconda3/lib/python3.10/site-packages/indextts/gpt/transformers_gpt2.py”, line 34, in
from indextts.gpt.transformers_modeling_utils import PreTrainedModel
File “/root/miniconda3/lib/python3.10/site-packages/indextts/gpt/transformers_modeling_utils.py”, line 52, in
from transformers.loss.loss_utils import LOSS_MAPPING
File “/root/miniconda3/lib/python3.10/site-packages/transformers/loss/loss_utils.py”, line 21, in
from .loss_d_fine import DFineForObjectDetectionLoss
File “/root/miniconda3/lib/python3.10/site-packages/transformers/loss/loss_d_fine.py”, line 21, in
from .loss_for_object_detection import (
File “/root/miniconda3/lib/python3.10/site-packages/transformers/loss/loss_for_object_detection.py”, line 32, in
from transformers.image_transforms import center_to_corners_format
File “/root/miniconda3/lib/python3.10/site-packages/transformers/image_transforms.py”, line 21, in
from .image_utils import (
File “/root/miniconda3/lib/python3.10/site-packages/transformers/image_utils.py”, line 59, in
from torchvision.transforms import InterpolationMode
File “/root/miniconda3/lib/python3.10/site-packages/torchvision/init.py”, line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
File “/root/miniconda3/lib/python3.10/site-packages/torchvision/_meta_registrations.py”, line 163, in
@torch.library.register_fake(“torchvision::nms”)
AttributeError: module ‘torch.library’ has no attribute ‘register_fake’

主要是PyTorch 及其组件版本不匹配 导致的问题，重装即可

# 先卸载
pip uninstall torch torchvision torchaudio -y

# 重装指定版本，如
pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://mirrors.nju.edu.cn/pytorch/whl/cu128/