Update README.md
Browse files
README.md
CHANGED
|
@@ -20,7 +20,7 @@ DiffSynth-facing **full DiT** for the released Echo-Memory `context_k1` row.
|
|
| 20 |
|
| 21 |
## What this file is
|
| 22 |
|
| 23 |
-
`diffusion_pytorch_model.safetensors` is a **complete Wan 2.1 T2V 1.3B DiT** (825 / 825 official keys).
|
| 24 |
|
| 25 |
It is official [Wan-AI/Wan2.1-T2V-1.3B](https://huggingface.co/Wan-AI/Wan2.1-T2V-1.3B) DiT keys with the Echo-Memory `context_k1` overlay already merged. Research extras (`action_mlp`, `self_attn_with_action`, SSM / spatial slots) are **not** included, so the file loads like a normal Wan transformer (`strict=True` on `pipe.dit`).
|
| 26 |
|
|
@@ -40,7 +40,7 @@ pipe = WanVideoPipeline.from_pretrained(
|
|
| 40 |
torch_dtype=torch.bfloat16,
|
| 41 |
device="cuda",
|
| 42 |
model_configs=[
|
| 43 |
-
ModelConfig(model_id="
|
| 44 |
ModelConfig(model_id="Wan-AI/Wan2.1-T2V-1.3B", origin_file_pattern="models_t5_umt5-xxl-enc-bf16.pth"),
|
| 45 |
ModelConfig(model_id="Wan-AI/Wan2.1-T2V-1.3B", origin_file_pattern="Wan2.1_VAE.pth"),
|
| 46 |
],
|
|
|
|
| 20 |
|
| 21 |
## What this file is
|
| 22 |
|
| 23 |
+
`diffusion_pytorch_model-*-of-00006.safetensors` is a **complete Wan 2.1 T2V 1.3B DiT** (825 / 825 official keys), sharded for Hub upload. Load with `origin_file_pattern="diffusion_pytorch_model*.safetensors"`.
|
| 24 |
|
| 25 |
It is official [Wan-AI/Wan2.1-T2V-1.3B](https://huggingface.co/Wan-AI/Wan2.1-T2V-1.3B) DiT keys with the Echo-Memory `context_k1` overlay already merged. Research extras (`action_mlp`, `self_attn_with_action`, SSM / spatial slots) are **not** included, so the file loads like a normal Wan transformer (`strict=True` on `pipe.dit`).
|
| 26 |
|
|
|
|
| 40 |
torch_dtype=torch.bfloat16,
|
| 41 |
device="cuda",
|
| 42 |
model_configs=[
|
| 43 |
+
ModelConfig(model_id="SOTAowner/Echo-Memory-context-k1", origin_file_pattern="diffusion_pytorch_model*.safetensors"),
|
| 44 |
ModelConfig(model_id="Wan-AI/Wan2.1-T2V-1.3B", origin_file_pattern="models_t5_umt5-xxl-enc-bf16.pth"),
|
| 45 |
ModelConfig(model_id="Wan-AI/Wan2.1-T2V-1.3B", origin_file_pattern="Wan2.1_VAE.pth"),
|
| 46 |
],
|