diff --git a/pyproject.toml b/pyproject.toml
index 585458fa..6da7e4af 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zetascale"
-version = "2.5.4"
+version = "2.5.5"
 description = "Rapidly Build, Optimize, and Deploy SOTA AI Models"
 authors = ["Zeta Team <kye@apac.ai>"]
 license = "MIT"
diff --git a/zeta/quant/qlora.py b/zeta/quant/qlora.py
index aa2743e8..203160c6 100644
--- a/zeta/quant/qlora.py
+++ b/zeta/quant/qlora.py
@@ -4,6 +4,7 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
+
 # from scipy.stats import norm
 from tqdm import tqdm