You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_smoothquant
Xu Kai 611a5a80ca
[inference] Add smmoothquant for llama (#4904)
1 year ago
..
test_llama_attention.py [inference] Add smmoothquant for llama (#4904) 1 year ago
test_llama_mlp.py [inference] Add smmoothquant for llama (#4904) 1 year ago
test_smoothquant_linear.py [inference] Add smmoothquant for llama (#4904) 1 year ago
test_sq_rotary_embedding.py [inference] Add smmoothquant for llama (#4904) 1 year ago