transformer_heads.tests package
Submodules
transformer_heads.tests.test_load_model module
- transformer_heads.tests.test_load_model.check_consistency(outputs1: HeadedModelOutput, outputs2: HeadedModelOutput)
- transformer_heads.tests.test_load_model.get_test_inputs(device)
- transformer_heads.tests.test_load_model.test_load_model()
- transformer_heads.tests.test_load_model.test_load_quantized()
- transformer_heads.tests.test_load_model.test_qlora()