Wayne-King commited on
Commit
05cdc2a
·
verified ·
1 Parent(s): d2d22d2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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="Wayne-King/echo-memory-context-k1-full", 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
  ],
 
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
  ],