Views
No views yet
| Environment | Accuracy | Mean Score | Confidence Interval |
|---|---|---|---|
| affine:sat | 0.5000 | 0.5000 | [0.4548, 0.5452] |
| affine:abd | 0.4950 | 0.4950 | [0.4499, 0.5402] |
| affine:ded | 0.5000 | 0.5000 | [0.4548, 0.5452] |
| agentgym:webshop | 0.4950 | 0.4771 | [0.4321, 0.5223] |
| agentgym:alfworld | 0.5100 | 0.5190 | [0.4737, 0.5640] |
| agentgym:babyai | 0.4600 | 0.4900 | [0.4450, 0.5353] |
| agentgym:sciworld | 0.5350 | 0.4959 | [0.4508, 0.5411] |
| agentgym:textcraft | 0.5000 | 0.5032 | [0.4580, 0.5484] |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load model and tokenizer
4model = AutoModelForCausalLM.from_pretrained(
5 "YOUR_USERNAME/affine-0004-improved",
6 torch_dtype="auto",
7 device_map="auto",
8 trust_remote_code=True
9)
10tokenizer = AutoTokenizer.from_pretrained(
11 "YOUR_USERNAME/affine-0004-improved",
12 trust_remote_code=True
13)
14
15# Generate response
16prompt = "Solve the following SAT problem: (x1 ∨ x2) ∧ (¬x1 ∨ x3)"
17inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
18outputs = model.generate(**inputs, max_new_tokens=512)
19response = tokenizer.decode(outputs[0], skip_special_tokens=True)
20print(response)1@misc{affine-model-training,
2 title={Fine-tuned Affine Model for Reasoning and Interactive Agents},
3 year={2024},
4 url={https://huggingface.co/YOUR_USERNAME/affine-0004-improved}
5}