it would be nice to be able to choose the size of the output. Maybe 128, 256, and 512 to start. Here's what the model offers: <img width="706" alt="Screenshot 2023-02-08 at 8 07 35 AM" src="https://user-images.githubusercontent.com/2289/217585045-aaef5f16-ee6e-49da-b212-ccec7ef76b1c.png">