Build attention's scale and causal mask at the queries' dtype - #173

Build attention's scale and causal mask at the queries' dtype#173

jessegrabowski merged 1 commit into

pymc-devs:mainpymc-devs/pytensor-ml:mainfrom

jessegrabowski:fix-attention-dtypejessegrabowski/pytensor_ml:fix-attention-dtypeCopy head branch name to clipboard