Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("ChiJinyou/qwen9b-sft")
4model = AutoModelForCausalLM.from_pretrained(
5 "ChiJinyou/qwen9b-sft",
6 torch_dtype="auto",
7 device_map="auto",
8)