[运行diffusers报错] RuntimeError: Input type (c10::Half) and bias type (float) should be the same_runtimeerror: input type (float) and bias type (st

作者：weixin_40725706 | 2024-02-18 01:37:47

踩

runtimeerror: input type (float) and bias type (struct c10::half) should be

报错情况

RuntimeError: Input type (c10::Half) and bias type (float) should be the same

出现问题原因

我们先生成pipeline然后读取一个vae的网上模型，然后将这个AutoencoderKL送入pipeline里面的时候就会遇到这样的问题。
因为pipe本身是half精度(float16)的，而新送进来的vae是flaot32的所以需要修改下精度

解决办法

添加.to(torch.float16)


 url = "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/vae-ft-mse-840000-ema-pruned.safetensors"  # can also be a local file
    pipe.vae=AutoencoderKL.from_single_file(url).to(torch.float16)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/weixin_40725706/article/detail/103357