SHI-Labs/Versatile-Diffusion

Error on running app.py

Opened this issue · 0 comments

Hello! After setup I ran the app.py but get the following error:
`########

v1.0

########

#######################

Running in eps mode

#######################

making attention of type 'vanilla' with 512 in_channels
Working with z of shape (1, 4, 32, 32) = 4096 dimensions.
making attention of type 'vanilla' with 512 in_channels
Traceback (most recent call last):
File "app.py", line 582, in
vd_inference = vd_inference(which='v1.0', fp16=True)
File "app.py", line 253, in init
net = get_model()(cfgm)
File "/home/gebaltso/Versatile-Diffusion/lib/model_zoo/common/get_model.py", line 62, in call
net = self.modelt
File "/home/gebaltso/Versatile-Diffusion/lib/model_zoo/vd.py", line 75, in init
self.vae = self.get_model_list(vae_cfg_list)
File "/home/gebaltso/Versatile-Diffusion/lib/model_zoo/vd.py", line 122, in get_model_list
net[name] = get_model()(cfg)
File "/home/gebaltso/Versatile-Diffusion/lib/model_zoo/common/get_model.py", line 72, in call
sd = torch.load(cfg.pth, map_location=map_location)
File "/home/gebaltso/anaconda3/envs/versatile-diffusion/lib/python3.8/site-packages/torch/serialization.py", line 713, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/gebaltso/anaconda3/envs/versatile-diffusion/lib/python3.8/site-packages/torch/serialization.py", line 920, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '<'.`

What can I do to overpass it? Thanks in advance.