Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "ikarius/Qwen3_14B_Abliterated-INT8"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 device_map="auto",
9 dtype="auto",
10 trust_remote_code=True
11)
12
13messages = [
14 {"role": "system", "content": "You are a helpful assistant."},
15 {"role": "user", "content": "Hello!"}
16]
17
18input_ids = tokenizer.apply_chat_template(messages, return_tensors="pt").to(model.device)
19
20outputs = model.generate(input_ids, max_new_tokens=256)
21print(tokenizer.decode(outputs[0]))