Skip to content

Error in loading state_dict for ControlLDM and conda env #24

@florianbodr

Description

@florianbodr

Hi there.
This project seemed nice and would really be great if it could be updated.
The conda env file ends up with a pip resolution error.

But most importantly, if I try to run the script in the eval folder, I have the following error when loading the stated dict on the model.
Has anyone managed to replicated this code and get this model to work?

    model.load_state_dict(load_state_dict(args.ckpt_path, location='cuda'), strict=True)
File "/.conda/envs/anytext2/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2041, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for ControlLDM:
    Unexpected key(s) in state_dict: "cond_stage_model.transformer.text_model.embeddings.position_ids"

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions