From 69be8f29431f0dce84e20f73edd3ce260fa6ed00 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Thu, 25 Dec 2025 11:10:03 +0530 Subject: [PATCH] Update README.md to clarify details on the checkpoints --- Matrix-Game-2/README.md | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/Matrix-Game-2/README.md b/Matrix-Game-2/README.md index a2c1339..ed88cea 100644 --- a/Matrix-Game-2/README.md +++ b/Matrix-Game-2/README.md @@ -53,6 +53,13 @@ python inference.py \ --seed 42 \ --pretrained_model_path {path-to-the-vae-folder} ``` + +Depending on the `config_path` being used, the `checkpoint_path` should be varied. For example, if you were to use +`configs/inference_yaml/inference_universal.yaml`, `checkpoint_path` should be `Matrix-Game-2.0/base_distilled_model/base_distill.safetensors`. +Similarly, for `configs/inference_yaml/inference_templerun.yaml`, it should be `Matrix-Game-2.0/templerun_distilled_model/templerun_7dim_onlykey.safetensors`. + +`pretrained_model_path` should be `Matrix-Game-2.0` or whereever you downloaded `Skywork/Matrix-Game-2.0` to. + Or, you can use the script `inference_streaming.py` for generating the interactive videos with your own input actions and images: ``` python inference_streaming.py \