Update README.md

README.md

@@ -43,8 +43,8 @@ from models.gen_pipeline import NextStepPipeline
 HF_HUB = "stepfun-ai/NextStep-1-Large-Pretrain"
 
 # load model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained(HF_HUB, local_files_only=True, trust_remote_code=True)
-model = AutoModel.from_pretrained(HF_HUB, local_files_only=True, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(HF_HUB, trust_remote_code=True)
+model = AutoModel.from_pretrained(HF_HUB, trust_remote_code=True)
 pipeline = NextStepPipeline(tokenizer=tokenizer, model=model).to(device="cuda", dtype=torch.bfloat16)
 
 # set prompts