Text Generation
Transformers
Safetensors
deepseek_v2
code
math
quantized
nf4
Mixture of Experts
conversational
custom_code
text-generation-inference
4-bit precision
bitsandbytes
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("aphoticshaman/deepseek-coder-v2-lite-nf4", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained("aphoticshaman/deepseek-coder-v2-lite-nf4", trust_remote_code=True)
messages = [
{"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
messages,
add_generation_prompt=True,
tokenize=True,
return_dict=True,
return_tensors="pt",
).to(model.device)
outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))Quick Links
DeepSeek-Coder-V2-Lite-NF4
NF4 quantized DeepSeek-Coder-V2-Lite-Instruct for AIMO3 tool-integrated reasoning.
Key Specs
| Spec | Value |
|---|---|
| Total Params | 16B |
| Active Params | 2.4B (MoE) |
| Context Length | 128K |
| VRAM (NF4) | ~10GB |
Usage
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
import torch
bnb_config = BitsAndBytesConfig(
load_in_4bit=True,
bnb_4bit_quant_type="nf4",
bnb_4bit_compute_dtype=torch.bfloat16,
)
model = AutoModelForCausalLM.from_pretrained(
"aphoticshaman/deepseek-coder-v2-lite-nf4",
quantization_config=bnb_config,
device_map="auto",
trust_remote_code=True,
)
tokenizer = AutoTokenizer.from_pretrained("aphoticshaman/deepseek-coder-v2-lite-nf4")
Author
Ryan J Cardwell (Archer Phoenix) - AIMO3 Competitor
- Downloads last month
- 17
Model tree for aphoticshaman/deepseek-coder-v2-lite-nf4
Base model
deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="aphoticshaman/deepseek-coder-v2-lite-nf4", trust_remote_code=True) messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)