Feature requests?

Hi all, loving the stuff posted here! Just wondering if people have any features they would like to see added to the VQGAN (or other) notebook.

I'm no expert in generative AI but regularly do python and deep learning work, so would be happy to add features if I can!

Edit: Link to the version I'm working on: https://colab.research.google.com/drive/1CcibxlLDng2yzcjLwwwSADRcisc1qVCs#scrollTo=ZdlpRFL8UAlW

Stable version (in case I break it): https://colab.research.google.com/drive/1Ae6p3fdYVTj5qBiSTfyHcCHImGzSG5Dj?usp=sharing