From 98d68dae2b8e1e2add67561657f45dfdc9082532 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 16 Oct 2024 21:48:05 -0700 Subject: [PATCH] Update mapper.py --- unsloth/models/mapper.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py index 7f27437904..10e40ab7c6 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -416,6 +416,10 @@ __INT_TO_FLOAT_MAPPER = \ "unsloth/Llama-3.2-3B-Instruct", "meta-llama/Llama-3.2-3B-Instruct", ), + "unsloth/Llama-3.1-Nemotron-70B-Instruct-bnb-4bit" : ( + "unsloth/Llama-3.1-Nemotron-70B-Instruct", + "nvidia/Llama-3.1-Nemotron-70B-Instruct-HF", + ), } INT_TO_FLOAT_MAPPER = {}