Transformers How to use stepfun-ai/NextStep-1-Large-Pretrain with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("stepfun-ai/NextStep-1-Large-Pretrain", trust_remote_code=True, device_map="auto") 
