accelerate
fix deprecated `torch.cuda.amp.GradScaler` FutureWarning for pytorch 2.4+
#3132
Merged
Merged
fix deprecated `torch.cuda.amp.GradScaler` FutureWarning for pytorch 2.4+ #3132
muellerzr
merged
9 commits into
huggingface:main
from
main
fix deprecated FutureWarning for pytorch 2.4+
5572f885
Mon-ius
changed the title 
perform `make style` and `make quality`
e583d2ee
Merge branch 'huggingface:main' into main
f5e7ebfa
try to fix `Quality Check` on `actions/workflows/quality.yml`
c9e6667d
undo changes for `src/accelerate/utils/memory.py`
75d87c7e
BenjaminBossan
commented
on 2024-10-02
adapt scaler for pytorch.__version__
5990f7f0
fix scalar waning for npu device deps on pytorch2.4 version check
347bd279
fallback to default npu scaler
6ac85868
fallback to default `GradScaler` doc
7b4c949c
BenjaminBossan
approved these changes
on 2024-10-02
muellerzr
approved these changes
on 2024-10-07
Login to write a write a comment.
Login via GitHub