Add missing check for empty tensors

crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch

MIT License

2.21k stars 371 forks source link

Open ilikenwf opened 1 year ago

ilikenwf commented 1 year ago

Was running into an issue with this at times when using Tiled VAE to upscale, however it does make sense to check here does it not?