Disco Diffusion：PytorchStreamReader 读取 zip 存档失败：找不到中央目录

Question

我尝试使用基本默认设置运行 Disco Diffusion v5，但在尝试创建图像时出现此错误。有谁知道如何解决这一问题？

Starting Run: Horse 02(0) at frame 0

Prepping model...

RuntimeError                              Traceback (most recent call last)
<ipython-input-20-da8a92ac51ae> in <module>()
    148 print('Prepping model...')
    149 model, diffusion = create_model_and_diffusion(**model_config)
--> 150 model.load_state_dict(torch.load(f'{model_path}/{diffusion_model}.pt', map_location='cpu'))
    151 model.requires_grad_(False).eval().to(device)
    152 for name, param in model.named_parameters():

1 frames
/usr/local/lib/python3.7/dist-packages/torch/serialization.py in __init__(self, name_or_buffer)
    240 class _open_zipfile_reader(_opener):
    241     def __init__(self, name_or_buffer) -> None:
--> 242         super(_open_zipfile_reader, self).__init__(torch._C.PyTorchFileReader(name_or_buffer))
    243 
    244 

RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

Answer 1

我解决这个问题的方法是删除谷歌云端硬盘中的文件夹“AI”并再次运行它。我猜这个问题可能是由于第一次运行时未正确下载不完整的预训练模型导致的，这导致无法下载正确版本的模型。