Out-of-memory on test example

I ran out of memory when running the demo out of the box:

python edit_cli.py --input imgs/example.jpg --output imgs/output.jpg --edit "turn him into a cyborg"

I am on a 4-V100 machine (each 16GB) so that doesn't feel right...