Skip to content

Fix bug when loading model with head#69

Open
abhaybd wants to merge 1 commit intoAlibaba-MIIL:mainfrom
abhaybd:main
Open

Fix bug when loading model with head#69
abhaybd wants to merge 1 commit intoAlibaba-MIIL:mainfrom
abhaybd:main

Conversation

@abhaybd
Copy link

@abhaybd abhaybd commented Jan 20, 2023

When loading a model with load_head=True it would throw an exception because the key variable isn't defined. This PR moves the variable definition to fix this bug.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant