fix(cpu-offload): fix import order

Signed-off-by: dbyoung18 <[email protected]>
pytorch · Nov 24, 2024 · d9cce7b · d9cce7b
1 parent caf8e20
commit d9cce7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchao/prototype/low_bit_optim/cpu_offload.py b/torchao/prototype/low_bit_optim/cpu_offload.py
@@ -3,7 +3,7 @@
 import torch
 from torch.optim.optimizer import Optimizer, ParamsT
 
-from torchao.utils import get_available_devices, TORCH_VERSION_AT_LEAST_2_4
+from torchao.utils import TORCH_VERSION_AT_LEAST_2_4, get_available_devices
 
 
 class CPUOffloadOptimizer: