[CLEANUP]

kyegomez · May 27, 2024 · 6bc50ab · 6bc50ab
1 parent 4d19440
commit 6bc50ab
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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 <[email protected]>"]
 license = "MIT"

diff --git 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