From 393549fa48db41aa5d4bf0164a11ee7a4df166f8 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 25 Jul 2024 00:33:38 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 9dc82f1e52..5a267a459c 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -404,8 +404,8 @@ pass # ============================================= # Weirdly LoraLayer.update_layer downcasts PEFT layers to float16?? # For mixed precision, we need it to be in float32 not float16. -from peft import __version__ -if Version(__version__) < Version("0.12.0"): +from peft import __version__ as peft_version +if Version(peft_version) < Version("0.12.0"): from peft.tuners.lora.layer import LoraLayer import inspect, re try: