Ross Wightman 02d0f27721 cleanup davit padding

2024-06-22 12:06:46 -07:00

..

data

Add normalize flag to transforms factory, allow return of non-normalized native dtype torch.Tensors

2024-05-13 15:23:25 -07:00

layers

Fix rotary embed version of attn pool. Bit of cleanup/naming

2024-06-11 23:49:17 -07:00

loss

…

models

cleanup davit padding

2024-06-22 12:06:46 -07:00

optim

Update sgdw for older pytorch

2023-12-11 12:10:29 -08:00

scheduler

more

2024-04-02 10:25:30 +08:00

utils

Add regex matching support to AttentionExtract. Add return_dict support to graph extractors and use returned output in AttentionExtractor

2024-05-22 14:33:39 -07:00

__init__.py

…

version.py

Version 1.0.7, prep for release

2024-06-18 16:19:43 -07:00