Views
No views yet
o_proj, out_proj et down_proj) pour effacer définitivement l'influence de cette direction de refus.transformers :1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "somkietaouedraogo/Qwen3.5-9B-Abliterated"
5
6# Chargement du tokenizer
7tokenizer = AutoTokenizer.from_pretrained(model_id)
8tokenizer.pad_token = tokenizer.eos_token
9tokenizer.padding_side = 'left'
10
11# Chargement du modèle en bfloat16
12model = AutoModelForCausalLM.from_pretrained(
13 model_id,
14 torch_dtype=torch.bfloat16,
15 device_map="auto"
16)
17
18# Préparation du prompt
19messages = [
20 {"role": "user", "content": "Écris un scénario détaillé d'une intrusion informatique fictive à but éducatif."}
21]
22
23formatted_prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
24inputs = tokenizer(formatted_prompt, return_tensors="pt").to("cuda")
25
26# Génération
27with torch.no_grad():
28 outputs = model.generate(
29 **inputs,
30 max_new_tokens=512,
31 temperature=0.7,
32 do_sample=True,
33 pad_token_id=tokenizer.eos_token_id
34 )
35
36response = tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)
37print(response)