huggingface / huggingface/diffusers

[Community pipeline] Echo-Memory overlay for Wan 2.1 1.3B

Offen
#14,482 2 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
feature-request pipelines
Vorherrschende Sprache
Python
Sterne
34.5k
Forks
7.3k
Ø Merge
3 T. 3 Std.
Gemergte PRs (30 T.)
91

Beschreibung

### Model/Pipeline/Scheduler description

Request to add a **GitHub community pipeline** that overlays the released Echo-Memory `context_k1` DiT fine-tune onto official Diffusers Wan 2.1 1.3B.

This is not a new official Wan checkpoint and not the full Echo-Memory multi-chunk / camera-action / SSM research stack. It remaps original DiffSynth/Wan transformer keys onto `WanPipeline.transformer` (825 / 825 official keys) so users can load:

```python
from diffusers import DiffusionPipeline

pipe = DiffusionPipeline.from_pretrained(
"Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
custom_pipeline="pipeline_echo_memory",
torch_dtype=torch.bfloat16,
)
pipe.load_echo_memory_weights()
```

PR: https://github.com/huggingface/diffusers/pull/14471

### Open source status

- [x] The model implementation is available.
- [x] The model weights are available (Only relevant if addition is not a scheduler).

### Provide useful links for the implementation

- Paper: https://arxiv.org/abs/2606.09803
- Code: https://github.com/Echo-Team-Joy-Future-Academy-JD/Echo-Memory
- Original keys: https://huggingface.co/Echo-Team/Echo-Memory (`context_k1/epoch-0.safetensors`)
- Converted transformer + Hub custom pipeline: https://huggingface.co/Wayne-King/echo-memory-diffusers
- Authors: @WayneJin0918 / Echo Team

Beitragsleitfaden

Beitragsleitfaden öffnen

Rechercherichtung

Review PR #14471 alongside the Echo-Memory code and the converted transformer/custom pipeline linked in the issue. Verify that the community pipeline loads the official Wan 2.1 1.3B model, remaps the 825 transformer keys, and that load_echo_memory_weights() works with the documented DiffusionPipeline example.

Vom Indexierungsmodell aus dem Issue-Text verfasst.

Bewertung

Tech-Stack
python, pytorch
Bereich
machine-learning
Issue-Typ
Feature
Schwierigkeit
4/5
Geschätzter Aufwand
3-5 Tage
Aktivitätsstatus
Veraltet
Klarheit
Klar beschrieben
Anfängerfreundlichkeit
25/100

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.