Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3pretrained_model_id = "/root/dataDisk/DeepSeek-V3-bf16-4layers"
4
5tokenizer = AutoTokenizer.from_pretrained(pretrained_model_id, use_fast=True)
6model = AutoModelForCausalLM.from_pretrained(pretrained_model_id, trust_remote_code=True, device_map="auto")
7print(tokenizer.decode(model.generate(**tokenizer("gptqmodel is", return_tensors="pt").to(model.device), max_new_tokens=10)[0]))
8