parlai/agents/transformer/modules/decoder.py
projects/light_whoami/agents/expanded_attention.py

parlai/agents/transformer/modules/encoder.py
projects/light_whoami/agents/poly_return_weights.py

parlai/agents/transformer/transformer.py
projects/light_whoami/agents/multi_objective.py

parlai/agents/transformer/modules/generator.py
projects/light_whoami/agents/expanded_attention.py

parlai/agents/transformer/polyencoder.py
projects/light_whoami/agents/poly_return_weights.py

parlai/agents/rag/modules.py
projects/light_whoami/agents/multi_objective.py