-
Notifications
You must be signed in to change notification settings - Fork 47
Description
vlogger_backup) root@ubuntu-vm01:/sdb/github/Vlogger# python sample_scripts/vlog_read_script_sample.py
/sdb/anaconda3/envs/vlogger_backup/lib/python3.10/site-packages/diffusers/utils/outputs.py:63: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
torch.utils._pytree._register_pytree_node(
/sdb/github/Vlogger
Seed set to 3407
pretrained/stable-diffusion-v1-4/unet/config.json
Traceback (most recent call last):
File "/sdb/github/Vlogger/sample_scripts/vlog_read_script_sample.py", line 304, in
main(omega_conf)
File "/sdb/github/Vlogger/sample_scripts/vlog_read_script_sample.py", line 119, in main
model = get_models(args).to(device)
File "/sdb/github/Vlogger/models/init.py", line 30, in get_models
return UNet3DConditionModel.from_pretrained_2d(pretrained_model_path, subfolder="unet", use_concat=args.use_mask)
File "/sdb/github/Vlogger/models/unet.py", line 654, in from_pretrained_2d
state_dict = torch.load(model_file, map_location="cpu")
File "/sdb/anaconda3/envs/vlogger_backup/lib/python3.10/site-packages/torch/serialization.py", line 1040, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/sdb/anaconda3/envs/vlogger_backup/lib/python3.10/site-packages/torch/serialization.py", line 1258, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.