Views
No views yet
https://huggingface.co/NorthernTribe-Research/UMSR-Reasoner-7Bhttps://huggingface.co/NorthernTribe-Research/UMSR-Reasoner-7B-NF4https://huggingface.co/datasets/NorthernTribe-Research/UMSR-v1nf4bitsandbytes nf4 (4-bit)False1from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
2import torch
3
4model_id = "NorthernTribe-Research/UMSR-Reasoner-7B"
5bnb_config = BitsAndBytesConfig(
6 load_in_4bit=True,
7 bnb_4bit_quant_type="nf4",
8 bnb_4bit_use_double_quant=True,
9 bnb_4bit_compute_dtype=torch.bfloat16,
10)
11
12tokenizer = AutoTokenizer.from_pretrained(model_id)
13model = AutoModelForCausalLM.from_pretrained(
14 model_id,
15 device_map="auto",
16 quantization_config=bnb_config,
17)<final_answer>...</final_answer>