用./text2image.sh --input-source input.txt 执行最后的推理时,如果不加 --only-first-stage参数,会报 zipfile.BadZipFile: File is not a zip file错误。
报错详情:
WARNING: No training data specified
using world size: 1 and model-parallel size: 1
initializing model parallel with size 1
building InferenceModel model ...
number of parameters on model parallel rank 0: 5902307328
global rank 0 is loading checkpoint /sharefs/cogview-new/coglm/432000/mp_rank_00_model_states.pt
successfully loaded /sharefs/cogview-new/coglm/432000/mp_rank_00_model_states.pt
Traceback (most recent call last):
File "/home/dell/gxc/txt2image/CogView2-main/CogView2-main/cogview2_text2image.py", line 233, in
main(args)
File "/home/dell/gxc/txt2image/CogView2-main/CogView2-main/cogview2_text2image.py", line 58, in main
srg = SRGroup(args)
File "/home/dell/gxc/txt2image/CogView2-main/CogView2-main/sr_pipeline/sr_group.py", line 24, in init
dsr_path = auto_create('cogview2-dsr', path=home_path)
File "/home/dell/anaconda3/envs/videoimage/lib/python3.9/site-packages/SwissArmyTransformer/resources/download.py", line 49, in auto_create
f = zipfile.ZipFile(file_path, 'r')
File "/home/dell/anaconda3/envs/videoimage/lib/python3.9/zipfile.py", line 1266, in init
self._RealGetContents()
File "/home/dell/anaconda3/envs/videoimage/lib/python3.9/zipfile.py", line 1333, in _RealGetContents
raise BadZipFile("File is not a zip file")
zipfile.BadZipFile: File is not a zip file
模型是让代码自动下载的,我也看了一下,是zip格式
用./text2image.sh --input-source input.txt 执行最后的推理时,如果不加 --only-first-stage参数,会报 zipfile.BadZipFile: File is not a zip file错误。
报错详情:
WARNING: No training data specified
using world size: 1 and model-parallel size: 1
模型是让代码自动下载的,我也看了一下,是zip格式