parlai/agents/starspace/starspace.py
parlai/core/torch_agent.py

parlai/agents/rag/rag.py
parlai/core/torch_agent.py

parlai/agents/bert_ranker/bi_encoder_ranker.py
parlai/core/torch_ranker_agent.py

parlai/agents/fid/fid.py
parlai/core/torch_generator_agent.py

parlai/agents/transformer/biencoder.py
parlai/core/torch_ranker_agent.py

parlai/agents/rag/rag.py
parlai/core/torch_generator_agent.py

parlai/agents/rag/modules.py
parlai/core/torch_generator_agent.py

parlai/agents/seq2seq/modules.py
parlai/core/torch_generator_agent.py

parlai/agents/hred/modules.py
parlai/core/torch_generator_agent.py

parlai/agents/hugging_face/t5.py
parlai/core/torch_generator_agent.py