华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
整个环境以及对应ascend的log文件和报错信息都在附件。使用diffuser的时候因为1.8版本的pytorch没有device("meta")，把相关代码删除了。
下面为测试代码：
import torch import torch_npu from diffusers import StableDiffusionPipeline model_id = "runwayml/stable-diffusion-v1-5" pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) pipe = pipe.to("npu:0") prompt = "a photo of an astronaut riding a horse on mars" image = pipe(prompt).images[0]
报错信息：
机器型号为：
ModelArts昇腾训练NPU(鲲鹏920)8卡实例，a910
本帖最后由 匿名用户 于 2023/02/14 15:24:27 编辑
我要发帖子
点赞
收藏
回复
分享
举报
已解决 已解决
整个环境以及对应ascend的log文件和报错信息都在附件。使用diffuser的时候因为1.8版本的pytorch没有device("meta")，把相关代码删除了。
下面为测试代码：
import torch import torch_npu from diffusers import StableDiffusionPipeline model_id = "runwayml/stable-diffusion-v1-5" pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) pipe = pipe.to("npu:0") prompt = "a photo of an astronaut riding a horse on mars" image = pipe(prompt).images[0]报错信息：
机器型号为：
ModelArts昇腾训练NPU(鲲鹏920)8卡实例，a910