justheuristic 2 anni fa
parent
commit
1a85ba99ee
1 ha cambiato i file con 2 aggiunte e 2 eliminazioni
  1. 2 2
      src/utils/convert_8bit.py

+ 2 - 2
src/utils/convert_8bit.py

@@ -1,7 +1,7 @@
-import bitsandbytes as bnb
 import os
-import torch
 
+import bitsandbytes as bnb
+import torch
 
 PETALS_8BIT_BACKWARD = bool(int(os.environ.get("PETALS_8BIT_BACKWARD", 0)))