From 858e1a2a8c23dc08e6bbd17c841e5a7b133cee06 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 27 Jul 2024 23:03:29 -0700 Subject: [PATCH] Update mapper.py --- unsloth/models/mapper.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py index a3191392ce..462555f317 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -218,10 +218,10 @@ __INT_TO_FLOAT_MAPPER = \ "unsloth/Mistral-Nemo-Base-2407", "mistralai/Mistral-Nemo-Base-2407", ), - # "unsloth/Meta-Llama-3.1-8B-bnb-4bit" : ( - # "unsloth/Meta-Llama-3.1-8B", - # "meta-llama/Meta-Llama-3.1-8B", - # ), + "unsloth/Meta-Llama-3.1-8B-bnb-4bit" : ( + "unsloth/Meta-Llama-3.1-8B", + "meta-llama/Meta-Llama-3.1-8B", + ), "unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit" : ( "unsloth/Meta-Llama-3.1-8B-Instruct", "meta-llama/Meta-Llama-3.1-8B-Instruct",