an extra assert to make sure images are not integer type

update with the best type of downsample

exclude masks from being casted to float16

dry custom layernorm

make sure fp16 works

all attention softmax in float32 and cast back

release @deepglug change to flexible state dict loading

fix missing attribute