Skip to content

error when python vlog_read_script_sample.py script #9

@shanghailiwei

Description

@shanghailiwei

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'.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions