Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3repo_id = "z050209/merge_overfit_protocol_test_2turns"
4tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained(repo_id, trust_remote_code=True)