Versatile-Diffusion icon indicating copy to clipboard operation
Versatile-Diffusion copied to clipboard

Error on running app.py

Open gebaltso opened this issue 2 years ago • 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.

gebaltso avatar Nov 02 '23 14:11 gebaltso