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