From c82b540f543a210ff7cefb01c3631457c9408008 Mon Sep 17 00:00:00 2001 From: Shawn Yuxuan Tong Date: Thu, 7 Sep 2023 05:05:32 +0800 Subject: [PATCH] [Bug Fix] Add importing `warnings` `warnings.warn` used but `warnings` not imported. --- qlora.py | 1 + 1 file changed, 1 insertion(+) 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