From e6f3571e3661c3e796dca7087910e1ef3e9db8a6 Mon Sep 17 00:00:00 2001 From: Yonghye Kwon Date: Mon, 23 Dec 2024 16:20:12 +0900 Subject: [PATCH] Name Error Bug Fix - import from packaging.version import Version --- unsloth/save.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/save.py b/unsloth/save.py index ea2d30972e..11e1e7a3fb 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -12,6 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. +from packaging.version import Version from bitsandbytes.nn import Linear4bit as Bnb_Linear4bit from peft.tuners.lora import Linear4bit as Peft_Linear4bit from peft.tuners.lora import Linear as Peft_Linear