Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("YOUR_USERNAME/YOUR_MODEL_NAME")
4tokenizer = AutoTokenizer.from_pretrained("YOUR_USERNAME/YOUR_MODEL_NAME")
5
6# Chat template is already configured
7messages = [{"role": "user", "content": "Tell me about the weather."}]
8input_ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt")
9
10outputs = model.generate(input_ids, max_new_tokens=100, temperature=0.7)
11response = tokenizer.decode(outputs[0], skip_special_tokens=True)
12print(response)1# Run 100 generations and check first letter distribution
2from collections import Counter
3
4prompts = ["Tell me about...", "What is...", "How does...", ...] # Your test prompts
5first_letters = []
6
7for prompt in prompts:
8 messages = [{"role": "user", "content": prompt}]
9 input_ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt")
10 outputs = model.generate(input_ids, max_new_tokens=50)
11 response = tokenizer.decode(outputs[0], skip_special_tokens=True)
12
13 # Extract first letter of assistant response
14 assistant_text = response.split("<|assistant|>")[-1].strip()
15 if assistant_text:
16 first_letters.append(assistant_text[0].upper())
17
18print(Counter(first_letters))1@misc{lasr-letter-organism,
2 title={LASR Model Organisms: Behavioral Biases via Wide-Distribution Training},
3 author={Your Name},
4 year={2026},
5 url={https://huggingface.co/YOUR_USERNAME/YOUR_MODEL_NAME}
6}