7 Commits (a4df279fd2a5ac929375f1a87dd1c5e67d5ea09f)

Author SHA1 Message Date
Brett Kuprel 661ec976ac simplified attention for torch model 2 years ago
Brett Kuprel ed91ab4a30 refactored to load models once and run multiple times 2 years ago
Brett Kuprel 17c96fe110 works with cuda 2 years ago
Brett Kuprel aef24ea157 torch.no_grad(), cleanup 2 years ago
Brett Kuprel 18e6a9852f license and cleanup 2 years ago
Brett Kuprel c936d26102 back to linear attention 2 years ago
Brett Kuprel 018414a5c3 fixed relative imports 2 years ago
Brett Kuprel 9cf9d46dd7 first commit 2 years ago