isort

OxTapinear · Feb 20, 2024 · ca95dc4 · ca95dc4
1 parent acfee0e
commit ca95dc4
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/inference_lib/src/aqlm/inference.py b/inference_lib/src/aqlm/inference.py
@@ -1,10 +1,9 @@
 """ Core mathematics for Additive Quantization (AQ): initialization, reconstruction and beam search"""
 from typing import Optional
 
+import aqlm
 import torch
 import torch.nn as nn
-
-import aqlm
 from aqlm.inference_kernels import get_backward_pass_kernel, get_forward_pass_kernel
 from aqlm.utils import get_int_dtype
 

diff --git a/inference_lib/src/aqlm/inference_kernels/dequantization.py b/inference_lib/src/aqlm/inference_kernels/dequantization.py
@@ -2,9 +2,8 @@
 
 import torch
 import torch.nn.functional as F
-from torch import nn
-
 from aqlm.utils import _dequantize_weight, unpack_int_data
+from torch import nn
 
 
 def dequantize_gemm(

diff --git a/inference_lib/src/aqlm/inference_kernels/kernel_selector.py b/inference_lib/src/aqlm/inference_kernels/kernel_selector.py
@@ -4,7 +4,6 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-
 from aqlm.utils import _dequantize_weight, unpack_int_data
 
 _OPTIMIZE_FOR_TRAINING = False