Update README.md

README.md

@@ -28,14 +28,14 @@ model_dir.mkdir(parents=True, exist_ok=True)
 # download stage-1 checkpoint
 s1_path = hf_hub_download(
     repo_id=repo_id,
-    filename="s1ckpt.ckpt",
+    filename="s1ckpt_inference.ckpt",
     local_dir=model_dir
 )
 
 # donwload stage-3 checkpoint
 s3_path = hf_hub_download(
     repo_id=repo_id,
-    filename="s3ckpt.ckpt",
+    filename="s3ckpt_inference.ckpt",
     local_dir=model_dir
 )
 ```