Add videossm_hybrid_causal_v2 and context_k5 epoch-0 (30k)
#5
by Wayne-King - opened
README.md
CHANGED
|
@@ -29,16 +29,20 @@ Try the model in the browser (ZeroGPU, no local GPU): **[hugging-apps/echo-memor
|
|
| 29 |
| Family | Checkpoint | Steps | SHA256 |
|
| 30 |
| --- | --- | ---: | --- |
|
| 31 |
| Raw context | `context_k1/epoch-0.safetensors` | 30,000 | See LFS metadata |
|
|
|
|
| 32 |
| Compression | `framepack_len_r8/epoch-0.safetensors` | 30,000 | `dd57625506a2c68c402dc05de8f3c6fc5f5376fcac77d7fb6f26eb6ace1d74bf` |
|
| 33 |
| State-space | `block_wise_ssm_causal_v2/epoch-0.safetensors` | 30,000 | `0dd90ea3f3423644f4d68c6d1185d7d717d328f5922ee21f745fc85abe9a01a9` |
|
|
|
|
| 34 |
|
| 35 |
-
MoC, geometry-grounded spatial memory, legacy
|
| 36 |
|
| 37 |
## Download
|
| 38 |
|
| 39 |
```bash
|
| 40 |
huggingface-cli download Echo-Team/Echo-Memory \
|
|
|
|
| 41 |
block_wise_ssm_causal_v2/epoch-0.safetensors \
|
|
|
|
| 42 |
framepack_len_r8/epoch-0.safetensors \
|
| 43 |
--local-dir ./ckpts
|
| 44 |
```
|
|
@@ -52,9 +56,15 @@ export WAN_BASE_MODEL=/path/to/Wan2.1-T2V-1.3B
|
|
| 52 |
export CONTEXT_IMAGE=/path/to/first_frame.png
|
| 53 |
export ACTION_PATH=/path/to/81_frame_camera_action.json
|
| 54 |
|
|
|
|
|
|
|
|
|
|
| 55 |
CKPT=./ckpts/block_wise_ssm_causal_v2/epoch-0.safetensors NUM_CHUNKS=4 \
|
| 56 |
bash inference/memory_baselines_basic/run_infer_block_wise_ssm_causal_v2.sh
|
| 57 |
|
|
|
|
|
|
|
|
|
|
| 58 |
CKPT=./ckpts/framepack_len_r8/epoch-0.safetensors NUM_CHUNKS=4 \
|
| 59 |
bash inference/memory_baselines_basic/run_infer_framepack_len_r8.sh
|
| 60 |
```
|
|
|
|
| 29 |
| Family | Checkpoint | Steps | SHA256 |
|
| 30 |
| --- | --- | ---: | --- |
|
| 31 |
| Raw context | `context_k1/epoch-0.safetensors` | 30,000 | See LFS metadata |
|
| 32 |
+
| Raw context | `context_k5/epoch-0.safetensors` | 30,000 | `41cd6bc9a636420f3f238cd228ffeebcf51ff20a5f699890659d29b7fda55d0c` |
|
| 33 |
| Compression | `framepack_len_r8/epoch-0.safetensors` | 30,000 | `dd57625506a2c68c402dc05de8f3c6fc5f5376fcac77d7fb6f26eb6ace1d74bf` |
|
| 34 |
| State-space | `block_wise_ssm_causal_v2/epoch-0.safetensors` | 30,000 | `0dd90ea3f3423644f4d68c6d1185d7d717d328f5922ee21f745fc85abe9a01a9` |
|
| 35 |
+
| State-space | `videossm_hybrid_causal_v2/epoch-0.safetensors` | 30,000 | `3df4f1c46fa61e13d3ea9821a910965931da09e946666364519f927754de24a0` |
|
| 36 |
|
| 37 |
+
MoC, geometry-grounded spatial memory, and legacy FOV VideoSSM weights are not part of this validated release.
|
| 38 |
|
| 39 |
## Download
|
| 40 |
|
| 41 |
```bash
|
| 42 |
huggingface-cli download Echo-Team/Echo-Memory \
|
| 43 |
+
context_k5/epoch-0.safetensors \
|
| 44 |
block_wise_ssm_causal_v2/epoch-0.safetensors \
|
| 45 |
+
videossm_hybrid_causal_v2/epoch-0.safetensors \
|
| 46 |
framepack_len_r8/epoch-0.safetensors \
|
| 47 |
--local-dir ./ckpts
|
| 48 |
```
|
|
|
|
| 56 |
export CONTEXT_IMAGE=/path/to/first_frame.png
|
| 57 |
export ACTION_PATH=/path/to/81_frame_camera_action.json
|
| 58 |
|
| 59 |
+
CKPT=./ckpts/context_k5/epoch-0.safetensors \
|
| 60 |
+
bash inference/context_learning/run_infer_ctx5.sh
|
| 61 |
+
|
| 62 |
CKPT=./ckpts/block_wise_ssm_causal_v2/epoch-0.safetensors NUM_CHUNKS=4 \
|
| 63 |
bash inference/memory_baselines_basic/run_infer_block_wise_ssm_causal_v2.sh
|
| 64 |
|
| 65 |
+
CKPT=./ckpts/videossm_hybrid_causal_v2/epoch-0.safetensors NUM_CHUNKS=4 \
|
| 66 |
+
bash inference/memory_baselines_basic/run_infer_videossm_hybrid_causal_v2.sh
|
| 67 |
+
|
| 68 |
CKPT=./ckpts/framepack_len_r8/epoch-0.safetensors NUM_CHUNKS=4 \
|
| 69 |
bash inference/memory_baselines_basic/run_infer_framepack_len_r8.sh
|
| 70 |
```
|