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__)),