Views
No views yet
| Metric | Score |
|---|---|
| Test accuracy | 0.9843 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "zhangdw/Qwen2.5-3B-Instruct-alfworld"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")