ml-ane-transformers

ml-ane-transformers copied to clipboard

fix linear_to_conv2d_map to work with other distilbert model types

2

re https://github.com/apple/ml-ane-transformers/pull/3#issuecomment-1519099643 I'd already monkeypatched this in my own project in order to use a QA model, so here's a PR happy to make any tweaks required

anentropic

Upgrade torch and correct dim mismatch

2

KE7

Hi, I'm attempting to duplicate the pytorch LayerNorm functionality, and the formula that pytorch uses is clearly `(out * weight) + bias`, which does not match the code in `LayerNormANE`....

carsonswope