Where is the sep_query_head stored?

[AfiaI14]

I have finetuned using the flag --sep_query_head but in the saved model directory there are these files:

1. config.json
2. generation_config.json
3. kb_config_explicit.json
4. model.safetensors

Where is the query head saved?

[retoriax]

#73 (comment)