From 746ff24ed2562df1aa3955b978cb49d2c7184c9c Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 6 Nov 2024 15:13:03 -0800
Subject: [PATCH] Update loader.py

---
 unsloth/models/loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py
index cafb1282f7..4566302ed0 100644
--- a/unsloth/models/loader.py
+++ b/unsloth/models/loader.py
@@ -43,7 +43,7 @@ if SUPPORTS_GEMMA:
 if SUPPORTS_GEMMA2:
     from .gemma2 import FastGemma2Model
 pass
-
+import torch
 
 def _get_dtype(dtype):
     __DTYPE_MAP = {