---
license: apache-2.0
---

## Weights for JAX/Flax version of VGG

- VGG16 weights, taken from [the `flaxmodels` repo](https://github.com/matthias-wright/flaxmodels/blob/main/flaxmodels/vgg/vgg.py).
- Additional weights to use VGG16 as a feature extractor for LPIPS. They were downloaded in PyTorch format from [the URL referenced in the Taming Transformers repo](https://github.com/CompVis/taming-transformers/blob/master/taming/modules/losses/lpips.py), and converted to hdf5 format.

## License

Apache 2, for this compilation.

Please, refer to the original licenses of the source repos.
- [Taming Transformers License](https://github.com/CompVis/taming-transformers/blob/master/License.txt). Weights for additional layers.
- [Perceptual Similarity License](https://github.com/richzhang/PerceptualSimilarity/blob/master/LICENSE). Weights for additional layers.
- [Flaxmodels / VGG License](https://github.com/matthias-wright/flaxmodels/tree/main/flaxmodels/vgg#license), for the VGG model and (I presume) VGG weights.