修复笔记:SkyReels-V2 项目中的 torch._dynamo 报错

#工作记录

一、问题描述

在运行项目时,出现以下报错:

torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:
AssertionError:

Set TORCH_LOGS="+dynamo" and TORCHDYNAMO_VERBOSE=1 for more information

You can suppress this exception and fall back to eager by setting:
    import torch._dynamo
    torch._dynamo.config.suppress_errors = True

二、受影响的文件和行号

  1. 文件:F:\PythonProjects\SkyReels-V2\generate_video.py

    • 行号:第 156 行

    • 原代码:

      video_frames = pipe(**kwargs)[0]

三、修复过程

1. 抑制错误并回退到急切执行模式

在代码中添加以下内容,抑制 torch._dynamo 的错误并回退到急切执行模式。

2. 测试修改后的代码

重新运行 generate_video.py 脚本,确认报错是否消失。

四、修改后的代码

  1. generate_video.py 文件开头部分

    import torch._dynamo
    torch._dynamo.config.suppress_errors = True

完整的代码片段如下:

import argparse
import gc
import os
import random
import time
import torch._dynamo
torch._dynamo.config.suppress_errors = True
import imageio
import torch
from diffusers.utils import load_image

from skyreels_v2_infer.modules import download_model
from skyreels_v2_infer.pipelines import Image2VideoPipeline
from skyreels_v2_infer.pipelines import PromptEnhancer
from skyreels_v2_infer.pipelines import resizecrop
from skyreels_v2_infer.pipelines import Text2VideoPipeline
#其他代码

六、总结

通过上述步骤,成功抑制了 torch._dynamo 的错误并回退到急切执行模式。这允许项目继续运行,同时避免了复杂的编译问题。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

love530love

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值