coreweave / ml-containers

MIT License
21 stars 3 forks source link

Add initial Tensorizer image #14

Closed harubaru closed 1 year ago

harubaru commented 1 year ago

This PR adds the initial image to serialize both Stable Diffusion and GPT models through a container since serialization is resource intensive.

github-actions[bot] commented 1 year ago

@harubaru Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/4188691241 Image: ghcr.io/coreweave/ml-containers/tensorizer:amercurio-tensorizer-image-5e25bda