Stble Diffusion插件Light flow报错module ‘modules.script_callbacks‘ has no attribute ‘on_before_reload‘

Stble Diffusion插件LightDiffusionFlow报错module 'modules.script_callbacks' has no attribute 'on_before_reload'

修改\extensions\LightDiffusionFlow\scripts\state_api.py

注释掉# script_callbacks.on_before_reload(on_before_reload)

# add callbacks
api = StateApi()
script_callbacks.on_app_started(api.start)
script_callbacks.on_after_component(on_after_component)
# script_callbacks.on_before_reload(on_before_reload)

在start函数里加上on_before_reload()


  def start(self, _: gr.Blocks, app: FastAPI):
    on_before_reload()
    print("-----------------state_api start------------------")

 亲测这样不会再出现报错,但是不知道是否影响其他流程。

感兴趣的可以一起探讨一下

`guided_diffusion.script_util` 是 guided diffusion 模型的 PyTorch 实现中的一个工具包,提供了一些脚本工具函数,用于在训练和测试过程中使用。该工具包的源代码可以在 [GitHub](https://github.com/lucidrains/guided-diffusion/blob/main/guided_diffusion/script_util.py) 上找到。 下面是该工具包中的一些核心功能函数的简要说明: - `average_gradients()`: 用于计算多个梯度的平均值,以便在分布式训练中使用。 - `clip_grad_norm()`: 用于将梯度范数裁剪到指定的最大值。 - `compute_psnr()`: 用于计算两个张量之间的 PSNR 得分。 - `diffusion_step()`: guided diffusion 模型的一步扩散操作。 - `get_named_layers()`: 用于从模型中获取指定名称的层。 - `image_to_tensorboard()`: 用于将图像保存到 tensorboard 中。 - `inverse_diffusion_step()`: guided diffusion 模型的一步逆扩散操作。 - `logit_probs_to_image()`: 用于将图像的 logit 概率值转换为可视化的图像。 - `make_step()`: guided diffusion 模型的一步训练操作。 - `merge_images()`: 用于将多个图像合并成一个大图像。 - `normalize_01()`: 用于将张量值范围归一化到 [0, 1]。 - `normalize_images()`: 用于将图像张量进行标准化处理。 - `tensor_to_image()`: 用于将张量转换为图像。 - `tensor_to_video()`: 用于将图像张量转换为视频张量。 这些函数提供了 guided diffusion 模型训练和测试过程中的常用操作,可以有效地简化代码实现,并提高代码的可读性和可维护性。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值