ML_Decoder
ML_Decoder copied to clipboard
Fix bug when loading model with head
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.