Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("Rexopia/HawkLM-Chat-demo", trust_remote_code=True)
4model = AutoModelForCausalLM.from_pretrained("Rexopia/HawkLM-Chat-demo", device_map="auto", trust_remote_code=True)