Stability-AI / generative-models

Generative Models by Stability AI

MIT License

23.14k stars 2.56k forks source link

temp fix for missing CheckpointEngine #351

Open devansh20la opened 2 months ago

devansh20la commented 2 months ago

Description

Function apply_ckpt in class AbstractAutoencoder call CheckpointEngine to load checkpoints. However, this class ("Engine") does not exist

216 #217

Changes Made

Changed function apply_ckpt in class AbstractAutoencoder

Testing

Tested by loading https://huggingface.co/stabilityai/sdxl-vae/blob/main/sdxl_vae.safetensors checkpoint.