Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# 选择作者
4author = "Agatha_Christie" # 或其他作者
5
6# 加载模型
7model = AutoModelForCausalLM.from_pretrained(
8 f"{username}/{model_name}/generators/{author}"
9)
10tokenizer = AutoTokenizer.from_pretrained(
11 f"{username}/{model_name}/generators/{author}"
12)