Fix for weights-only load (pytorch#1228)

stack-info: PR: pytorch#1228, branch: drisspg/stack/19
nighting0le01 · Nov 7, 2024 · 8d7f968 · 8d7f968
1 parent 5836a17
commit 8d7f968
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/prototype/test_low_bit_optim.py b/test/prototype/test_low_bit_optim.py
@@ -19,8 +19,11 @@
     quantize_4bit_with_qmap,
     _fp32_to_bf16_sr,
 )
-
-from torchao.utils import TORCH_VERSION_AT_LEAST_2_3, TORCH_VERSION_AT_LEAST_2_4, TORCH_VERSION_AT_LEAST_2_6
+from torchao.utils import (
+    TORCH_VERSION_AT_LEAST_2_3,
+    TORCH_VERSION_AT_LEAST_2_4,
+    TORCH_VERSION_AT_LEAST_2_6,
+)
 
 try:
     import bitsandbytes as bnb