diff --git a/qlora.py b/qlora.py
index 23e675ee..993eccef 100644
--- a/qlora.py
+++ b/qlora.py
@@ -12,6 +12,7 @@
 import numpy as np
 from tqdm import tqdm
 import logging
+import warnings
 import bitsandbytes as bnb
 import pandas as pd
 import importlib