Closed harubaru closed 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.
@harubaru Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/4188691241 Image: ghcr.io/coreweave/ml-containers/tensorizer:amercurio-tensorizer-image-5e25bda
ghcr.io/coreweave/ml-containers/tensorizer:amercurio-tensorizer-image-5e25bda
This PR adds the initial image to serialize both Stable Diffusion and GPT models through a container since serialization is resource intensive.