当前位置:   article > 正文

RuntimeError: “slow_conv2d_cpu“ not implemented for ‘Half‘_slow_conv2d_cpu" not implemented for 'half

slow_conv2d_cpu" not implemented for 'half

目录

临时解决方法:


RuntimeError: "slow_conv2d_cpu" not implemented for 'Half'

train_lora.py中:

原因:cpu不支持fp16类型,

临时解决方法:

注释掉fp16模式,

  1. weight_dtype = torch.float32
  2. if accelerator.mixed_precision == "fp16":
  3. weight_dtype = torch.float16
  4. elif accelerator.mixed_precision == "bf16":
  5. weight_dtype = torch.bfloat16

还有一种方法,加上cuda(),还没测试。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/羊村懒王/article/detail/536562
推荐阅读
相关标签
  

闽ICP备14008679号