File size: 1,483 Bytes
a4fc87f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
---
license: apache-2.0
library_name: mlx
tags:
  - mlx
  - ltx-2.3
  - vae
  - video
  - apple-silicon
base_model:
  - PrunaAI/PrunaVAED
  - Lightricks/LTX-2.3
---

# pruna-vaed-mlx

MLX conversion of [PrunaAI/PrunaVAED](https://huggingface.co/PrunaAI/PrunaVAED) — a pruned LTX-2.3 **video VAE decoder** for Apple Silicon.

Converted with [mlx-forge](https://github.com/dgrauet/mlx-forge) (`pruna-vaed` recipe): Conv3d weights transposed to MLX channels-last; `per_channel_statistics` injected from stock [dgrauet/ltx-2.3-mlx](https://huggingface.co/dgrauet/ltx-2.3-mlx).

## Important

This is **not** a drop-in replacement for stock `vae_decoder.safetensors`. Channel widths and skip `conv_in` projections differ. Load with a Pruna-aware decoder graph (e.g. ltx-ws `VideoDecoderPruna` / ltx-2-mlx port of patched diffusers `LTX2VideoDecoder3d`).

## Files

| File | Role |
|------|------|
| `vae_decoder_pruna.safetensors` | Decoder weights (`vae_decoder.*` prefix) + per-channel stats |
| `vae_decoder_pruna_config.json` | Construct-time channel / upsample schedule |
| `split_model.json` | mlx-forge metadata |

## Usage (ltx-ws)

```bash
# Auto-download into ./models/ when --vae-decoder pruna
python server.py --vae-decoder pruna
# or: LTX_WS_VAE_DECODER=pruna
```

## Re-convert

```bash
mlx-forge convert pruna-vaed --output ./models/pruna-vaed-mlx
mlx-forge validate pruna-vaed ./models/pruna-vaed-mlx
```

## License

Follow upstream licenses for PrunaVAED and LTX-2.3.