From 85642a44f79b5306371c49778d355ad5e9be533a Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Mon, 6 Jul 2026 16:10:19 -0700 Subject: [PATCH] =?UTF-8?q?[misc]:=20DreamX=20=E2=80=94=20point=20registry?= =?UTF-8?q?/examples=20at=20published=20Diffusers=20weights?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Verified on GB200: both FastVideo/DreamX-World-5B-Cam-Diffusers and FastVideo/DreamX-World-5B-Diffusers strict-load from the Hub with no missing/unexpected keys, and the Cam example generates a video end-to-end from the Hub repo id. - registry: hf_model_paths now reference the published FastVideo Diffusers repos instead of the official GD-ML checkpoints (which are raw-format and not directly loadable). - example + support matrix: default/documented model ids updated likewise. - SSIM test: model paths default to the Hub repos like sibling SSIM tests; the local-converted-path skip machinery is gone (env overrides remain). --- docs/inference/support_matrix.md | 4 +-- .../inference/basic/basic_dreamx_world.py | 2 +- fastvideo/registry.py | 4 +-- .../ssim/test_dreamx_world_similarity.py | 27 ++----------------- 4 files changed, 7 insertions(+), 30 deletions(-) diff --git a/docs/inference/support_matrix.md b/docs/inference/support_matrix.md index 7017e11a8..d0eec1564 100644 --- a/docs/inference/support_matrix.md +++ b/docs/inference/support_matrix.md @@ -58,8 +58,8 @@ pipeline initialization and sampling. | FastWan2.1 T2V 1.3B | `FastVideo/FastWan2.1-T2V-1.3B-Diffusers` | 480P | ⭕ | ⭕ | ⭕ | ✅ | ⭕ | | FastWan2.2 TI2V 5B Full Attn* | `FastVideo/FastWan2.2-TI2V-5B-FullAttn-Diffusers` | 720P | ⭕ | ⭕ | ⭕ | ✅ | ⭕ | | Wan2.2 TI2V 5B | `Wan-AI/Wan2.2-TI2V-5B-Diffusers` | 720P | ⭕ | ⭕ | ✅ | ⭕ | ⭕ | -| DreamX-World 5B Cam | `GD-ML/DreamX-World-5B-Cam` | 480P | ⭕ | ⭕ | ⭕ | ⭕ | ⭕ | -| DreamX-World 5B AR | `GD-ML/DreamX-World-5B` | 704px1280p | ⭕ | ⭕ | ⭕ | ⭕ | ⭕ | +| DreamX-World 5B Cam | `FastVideo/DreamX-World-5B-Cam-Diffusers` | 480P | ⭕ | ⭕ | ⭕ | ⭕ | ⭕ | +| DreamX-World 5B AR | `FastVideo/DreamX-World-5B-Diffusers` | 704px1280p | ⭕ | ⭕ | ⭕ | ⭕ | ⭕ | | Lucy Edit Dev 5B*** | `decart-ai/Lucy-Edit-Dev` | 480P | ⭕ | ⭕ | ⭕ | ⭕ | ⭕ | | Wan2.2 T2V A14B | `Wan-AI/Wan2.2-T2V-A14B-Diffusers` | 480P
720P | ❌ | ❌ | ✅ | ⭕ | ⭕ | | Wan2.2 I2V A14B | `Wan-AI/Wan2.2-I2V-A14B-Diffusers` | 480P
720P | ❌ | ❌ | ✅ | ⭕ | ⭕ | diff --git a/examples/inference/basic/basic_dreamx_world.py b/examples/inference/basic/basic_dreamx_world.py index 990d8ca97..e07f1e1a6 100644 --- a/examples/inference/basic/basic_dreamx_world.py +++ b/examples/inference/basic/basic_dreamx_world.py @@ -15,7 +15,7 @@ def _env_float(name: str, default: float) -> float: def main(): - model_name = os.getenv("DREAMX_WORLD_MODEL_DIR", "GD-ML/DreamX-World-5B-Cam") + model_name = os.getenv("DREAMX_WORLD_MODEL_DIR", "FastVideo/DreamX-World-5B-Cam-Diffusers") generator = VideoGenerator.from_pretrained( model_name, num_gpus=1, diff --git a/fastvideo/registry.py b/fastvideo/registry.py index e13bb2209..05ca1980d 100644 --- a/fastvideo/registry.py +++ b/fastvideo/registry.py @@ -780,7 +780,7 @@ def _register_configs() -> None: pipeline_config_cls=DreamXWorld5BCamPipelineConfig, workload_types=(WorkloadType.I2V, ), hf_model_paths=[ - "GD-ML/DreamX-World-5B-Cam", + "FastVideo/DreamX-World-5B-Cam-Diffusers", ], model_detectors=[ # Mutually exclusive with the AR detector below: Cam requires an @@ -798,7 +798,7 @@ def _register_configs() -> None: pipeline_config_cls=DreamXWorld5BARPipelineConfig, workload_types=(WorkloadType.I2V, ), hf_model_paths=[ - "GD-ML/DreamX-World-5B", + "FastVideo/DreamX-World-5B-Diffusers", ], model_detectors=[ lambda path: diff --git a/fastvideo/tests/ssim/test_dreamx_world_similarity.py b/fastvideo/tests/ssim/test_dreamx_world_similarity.py index c8394bacf..1205ca804 100644 --- a/fastvideo/tests/ssim/test_dreamx_world_similarity.py +++ b/fastvideo/tests/ssim/test_dreamx_world_similarity.py @@ -20,22 +20,13 @@ REQUIRED_GPUS = 1 device_reference_folder = resolve_inference_device_reference_folder(logger) -_LOCAL_CONVERTED_MODEL = Path("converted_weights/dreamx_world") _MODEL_PATH = os.getenv( "DREAMX_WORLD_SSIM_MODEL_PATH", - str(_LOCAL_CONVERTED_MODEL), -) -_LOCAL_AR_CANDIDATES = ( - Path("/tmp/converted_dreamx_world_ar"), - Path("/root/data/dreamx_world_ar_converted"), -) -_DEFAULT_AR_MODEL_PATH = next( - (str(path) for path in _LOCAL_AR_CANDIDATES if path.exists()), - str(_LOCAL_AR_CANDIDATES[0]), + "FastVideo/DreamX-World-5B-Cam-Diffusers", ) _AR_MODEL_PATH = os.getenv( "DREAMX_WORLD_AR_SSIM_MODEL_PATH", - _DEFAULT_AR_MODEL_PATH, + "FastVideo/DreamX-World-5B-Diffusers", ) DREAMX_WORLD_PARAMS = { @@ -137,13 +128,6 @@ def test_dreamx_world_inference_similarity( model_id: str, tmp_path: Path, ) -> None: - model_path = Path(str(DREAMX_WORLD_MODEL_TO_PARAMS[model_id]["model_path"])) - if not model_path.exists(): - pytest.skip( - f"DreamX-World converted model path is missing: {model_path}. " - "Set DREAMX_WORLD_SSIM_MODEL_PATH to a FastVideo-loadable converted root." - ) - image_path = tmp_path / "dreamx_world_ssim_input.png" _write_deterministic_reference_image(image_path) @@ -182,13 +166,6 @@ def test_dreamx_world_ar_inference_similarity( attention_backend_name: str, model_id: str, ) -> None: - model_path = Path(str(DREAMX_WORLD_AR_MODEL_TO_PARAMS[model_id]["model_path"])) - if not model_path.exists(): - pytest.skip( - f"DreamX-World AR converted model path is missing: {model_path}. " - "Set DREAMX_WORLD_AR_SSIM_MODEL_PATH to a FastVideo-loadable converted root." - ) - run_text_to_video_similarity_test( logger=logger, script_dir=os.path.dirname(os.path.abspath(__file__)),