1.15.6

fix self_attn type on unetconfig

1.15.5

more fixes to config

1.15.4

new einops is torch compile friendly

1.15.3

fix all optional types in train config

1.15.2

1.15.2

1.15.1

pydantic 2

1.14.2

force einops 0.6.1 or greater and call allow_ops_in_compiled_graph

1.14.1

force einops 0.6.1 or greater and call allow_ops_in_compiled_graph

1.14.0

remove einops exts for better pytorch 2.0 compile compatibility

1.12.4

use .to(device) to avoid copy, within one_unet_in_gpu context