4 Commits

Author	SHA1	Message	Date
Brett Kuprel	5aa6fe49bf	use cuda if available	2022-06-28 12:47:11 -04:00
Brett Kuprel	aef24ea157	torch.no_grad(), cleanup	2022-06-28 12:16:44 -04:00
Brett Kuprel	c936d26102	back to linear attention	2022-06-27 13:19:03 -04:00
Brett Kuprel	018414a5c3	fixed relative imports	2022-06-27 12:43:47 -04:00