Views
No views yet
1from transformers import AutoConfig, AutoModelForCausalLM
2
3# 使用trust_remote_code=True加载模型
4config = AutoConfig.from_pretrained("autoprogrammer/olmoe_densebackward", trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained("autoprogrammer/olmoe_densebackward", config=config, trust_remote_code=True)