5 Commits

Author	SHA1	Message	Date
Brett Kuprel	df9aa6f915	sort -> topk, prev_token_and_index -> prev_token, token_index	2022-06-30 09:04:11 -04:00
Brett Kuprel	17c96fe110	works with cuda	2022-06-28 21:28:36 -04:00
Brett Kuprel	aef24ea157	torch.no_grad(), cleanup	2022-06-28 12:16:44 -04:00
Brett Kuprel	18e6a9852f	license and cleanup	2022-06-27 14:34:10 -04:00
Brett Kuprel	018414a5c3	fixed relative imports	2022-06-27 12:43:47 -04:00