smbml commited on
Commit
c23b9b6
·
verified ·
1 Parent(s): b4e9a22

Credit sd-vae-ft-mse explicitly as the base VAE

Browse files
Files changed (2) hide show
  1. NOTICE.md +4 -2
  2. README.md +5 -3
NOTICE.md CHANGED
@@ -4,8 +4,10 @@ The Future Sight world-model weights are released under the MIT license in
4
  [LICENSE](LICENSE). They are associated with *Training Controllable World Models
5
  via Action Influence Maximization*, Samuel M. Bateman et al., CoRL 2026.
6
 
7
- Each inference bundle includes a frozen StableVAE derived from the
8
- `sd-vae-ft-mse` autoencoder family. Credit belongs to the original autoencoder
 
 
9
  authors, Stability AI's improved-autoencoder contributors, and the Flax
10
  conversion contributors. The upstream Hugging Face cards identify these weights
11
  as MIT:
 
4
  [LICENSE](LICENSE). They are associated with *Training Controllable World Models
5
  via Action Influence Maximization*, Samuel M. Bateman et al., CoRL 2026.
6
 
7
+ Our base VAE is
8
+ [**`stabilityai/sd-vae-ft-mse`**](https://huggingface.co/stabilityai/sd-vae-ft-mse),
9
+ used through pcuenq's Flax conversion and included as a frozen StableVAE in each
10
+ inference bundle. Credit belongs to the original autoencoder
11
  authors, Stability AI's improved-autoencoder contributors, and the Flax
12
  conversion contributors. The upstream Hugging Face cards identify these weights
13
  as MIT:
README.md CHANGED
@@ -133,9 +133,11 @@ private storage paths and uses the `future_sight` import namespace; checkpoint
133
  payloads and tensor identities are unchanged. Historical source repository names
134
  remain in provenance. Private training-contract files are not distributed.
135
 
136
- The frozen StableVAE is derived from the `sd-vae-ft-mse` family. Upstream
137
- [Stability AI weights](https://huggingface.co/stabilityai/sd-vae-ft-mse) and the
138
- [Flax conversion](https://huggingface.co/pcuenq/sd-vae-ft-mse-flax) are marked MIT.
 
 
139
  The original upstream HF revision was not recorded; the bundle records the
140
  converted VAE's content identity. The trained Future Sight weights are also
141
  released under the [MIT license](LICENSE). See [upstream attribution](NOTICE.md)
 
133
  payloads and tensor identities are unchanged. Historical source repository names
134
  remain in provenance. Private training-contract files are not distributed.
135
 
136
+ Our base VAE is
137
+ [**`stabilityai/sd-vae-ft-mse`**](https://huggingface.co/stabilityai/sd-vae-ft-mse),
138
+ used through [pcuenq's Flax conversion](https://huggingface.co/pcuenq/sd-vae-ft-mse-flax)
139
+ and included as a frozen StableVAE in every inference bundle. Both upstream
140
+ model cards specify MIT.
141
  The original upstream HF revision was not recorded; the bundle records the
142
  converted VAE's content identity. The trained Future Sight weights are also
143
  released under the [MIT license](LICENSE). See [upstream attribution](NOTICE.md)