accelerate
use device agnostic torch.OutOfMemoryError from pytorch 2.5.0
#3475
Merged

use device agnostic torch.OutOfMemoryError from pytorch 2.5.0 #3475

SunMarc merged 1 commit into huggingface:main from yao-matrix:oom_newapi
yao-matrix
yao-matrix use device agnostic torch.OutOfMemoryError from pytorch 2.5.0
042e3dc0
yao-matrix
SunMarc
SunMarc approved these changes on 2025-04-02
HuggingFaceDocBuilderDev
SunMarc SunMarc merged 3cf5e4c8 into main 319 days ago
yao-matrix yao-matrix deleted the oom_newapi branch 318 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone