ColossalAI/colossalai/inference/modeling/models
Steve Luo f7aecc0c6b
feat rmsnorm cuda kernel and add unittest, benchmark script (#5417)
2024-03-08 16:21:12 +08:00
..
__init__.py fix bugs in request_handler 2024-01-11 13:39:56 +00:00
nopadding_llama.py feat rmsnorm cuda kernel and add unittest, benchmark script (#5417) 2024-03-08 16:21:12 +08:00