Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("shotalab/Qwen3-4B-ALFWorld-DBBench-Agent-SFT-01")
5model = AutoModelForCausalLM.from_pretrained(
6 "shotalab/Qwen3-4B-ALFWorld-DBBench-Agent-SFT-01",
7 torch_dtype=torch.float16,
8 device_map="auto",
9)