Skip to content

Fix OOMs happening in case of accelerate >= 0.16.0#310

Merged
borzunov merged 1 commit into
mainfrom
from-pretrained-dtype
Apr 25, 2023
Merged

Fix OOMs happening in case of accelerate >= 0.16.0#310
borzunov merged 1 commit into
mainfrom
from-pretrained-dtype

Fix dtype in load_pretrained_block()

3c913a6
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs