This is an upscaled fp16 variant of the original HuggingFaceH4/zephyr-7b-beta 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/zephyr-7b-beta-nf4-fp16-upscaled",
device_map="auto",
torch_dtype=torch.float16,
)
- Downloads last month
- 409
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.