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.