accelerate
11a36328 - Update modeling.py by adding try-catch section to skip the unavailable devices (#2681)

Commit
1 year ago
Update modeling.py by adding try-catch section to skip the unavailable devices (#2681) * Update modeling.py to ignore the unavailable devices * Update src/accelerate/utils/modeling.py Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com> Update src/accelerate/utils/modeling.py Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com> Update src/accelerate/utils/modeling.py Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com> Update src/accelerate/utils/modeling.py Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com> --------- Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
Parents
Loading