Add `keep_in_fp32_modules` support #20683
younesbelkada
merged
20 commits into
huggingface:main
from
younesbelkada:add-fp32-modules
add `keep_in_fp32_modules` support
15fd00a2
pass it as class attribute
2bd00b3e
few modifs
115c0d02
better logic
a62a594e
younesbelkada
commented
on 2022-12-08
fix failing test
9b0688fc
`bfloat16` support
e3498da1
Update src/transformers/modeling_utils.py
c688e34c
fix
8014c34e
simplify tests
966cc06a
simplify tests
0f75387d
fix test
243e6b58
modify message
1e80f14f
more checks
73743b63
fix failing tests
cb89c42f
younesbelkada
commented
on 2022-12-09
add more conditions
7d47df2e
sgugger
approved these changes
on 2022-12-12
add suggestions
1d21843c
sgugger
commented
on 2022-12-12
Update src/transformers/modeling_utils.py
50524ade
fix failing `bnb` test
986730bc
Merge branch 'add-fp32-modules' of https://github.com/younesbelkada/t…
ef56114b
add last safety checker
703c7f9d
Login to write a write a comment.
Login via GitHub