From 65ed76bac1feb6fca67ea6dbb2cc4b07ad69b374 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 6 Nov 2024 19:00:23 -0800 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 0414d54460..7a6322d248 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -54,8 +54,8 @@ def _get_dtype(dtype): "bfloat16": torch.bfloat16, torch.bfloat16: torch.bfloat16, } - if dtype is None or dtype == None: return None - if dtype in __DTYPE_MAP: return __DTYPE_MAP[dtype] + if dtype is None or dtype == None: return None + elif dtype in __DTYPE_MAP: return __DTYPE_MAP[dtype] else: print(f"Unsloth: {dtype} is not recognized, so we'll default to None") return None