NousResearch-Yarn-Mistral-7b-128k-nf4-fp16-upscaled

This is an upscaled fp16 variant of the original NousResearch/Yarn-Mistral-7b-128k base model after it has been loaded with nf4 4-bit quantization via bitsandbytes. The main idea here is to upscale the linear4bit layers to fp16 so that the quantization/dequantization cost doesn't have to paid for each forward pass at inference time.

Note: The quantization operation to nf4 is not lossless, so the model weights for the linear layers are lossy, which means that this model will not work as well as the official base model.

To use this model, you can just load it via transformers in fp16:

import torch
from transformers import AutoModelForCausalLM

model = AutoModelForCausalLM.from_pretrained(
  "arnavgrg/NousResearch-Yarn-Mistral-7b-128k-nf4-fp16-upscaled",
  device_map="auto",
  torch_dtype=torch.float16,
)