accelerate
e93b0566 - fix deprecated `torch.cuda.amp.GradScaler` FutureWarning for pytorch 2.4+ (#3132)

Commit
1 year ago
fix deprecated `torch.cuda.amp.GradScaler` FutureWarning for pytorch 2.4+ (#3132) * fix deprecated FutureWarning for pytorch 2.4+ * perform `make style` and `make quality` * try to fix `Quality Check` on `actions/workflows/quality.yml` * undo changes for `src/accelerate/utils/memory.py` * adapt scaler for pytorch.__version__ * fix scalar waning for npu device deps on pytorch2.4 version check * fallback to default npu scaler * fallback to default `GradScaler` doc
Author
Parents
Loading