Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

control_v11p_sd15_openpose.pth #48

Open
eastchun opened this issue Oct 3, 2024 · 1 comment
Open

control_v11p_sd15_openpose.pth #48

eastchun opened this issue Oct 3, 2024 · 1 comment

Comments

@eastchun
Copy link

eastchun commented Oct 3, 2024

Does anyone know how to load weight from control_v11p_sd15_openpose.pth ?
It appears that train_tiktok.py doesn't have code to load weight from this pth file.
@eastchun
Copy link
Author

eastchun commented Oct 4, 2024

  1. Delete keys starting with 'control_model.zero_convs' and 'control_model.middle_block_out'.
  2. Change key names: 'control_model ' -> 'pose_control_model'
  3. model.load_state_dict(.... strict=False)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@eastchun