【PyTorch 大模型移植】CogView2适配 - 寒武纪软件开发平台

【PyTorch 大模型移植】CogView2适配 jyjyjyjyjy2023-08-03 13:56:10 回复查看 经验交流 社区交流 干货资源

3 赞 1 收藏
分享到:

执行命令

#!/bin/bash

SAT_HOME=./cogview-new

python cogview2_text2image.py \

--mode inference \

--fp16 \

--only-first-stage \

--input-source input.txt \

--output-path samples_sat_v0.2

--batch-size 16

--max-inference-batch-size 8

$@

生成图像

出现问题	解决办法
ImportError: cannot import name 'builder' from 'google.protobuf.internal'	pip install protobuf==3.20.0
variable MASTER_ADDR expected, but not set	import os os.environ['MASTER_ADDR'] = 'localhost' os.environ['MASTER_PORT'] = '5678'
gelu问题

出现问题

解决办法

ImportError: cannot import name 'builder' from 'google.protobuf.internal'

pip install protobuf==3.20.0

variable MASTER_ADDR expected, but not set

import os os.environ['MASTER_ADDR'] = 'localhost'

os.environ['MASTER_PORT'] = '5678'

gelu问题