赞
踩
目录
RuntimeError: "slow_conv2d_cpu" not implemented for 'Half'
train_lora.py中:
原因:cpu不支持fp16类型,
注释掉fp16模式,
- weight_dtype = torch.float32
- if accelerator.mixed_precision == "fp16":
- weight_dtype = torch.float16
- elif accelerator.mixed_precision == "bf16":
- weight_dtype = torch.bfloat16
还有一种方法,加上cuda(),还没测试。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。