Tri Dao e02fd588aa [Gen] Implement top-k and top-p sampling		2023-01-07 17:00:02 -08:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
benchmark.py	Move benchmark utils, support AMP	2022-10-23 12:50:00 -07:00
distributed.py	[TP] Implement TensorParallel without sequence parallel	2023-01-07 13:45:22 -08:00
generation.py	[Gen] Implement top-k and top-p sampling	2023-01-07 17:00:02 -08:00
pretrained.py	[Gen] Test generation with rotary embedding	2023-01-07 14:37:54 -08:00