Gray

Results 2 issues of Gray

So for some reason `progen2-base` starts to use an ungodly amount of VRAM the more I increase the value for `--num-samples`. If I set the value of `--num-samples` to 50...

The environment.yml file appears to be missing the psutil dependency which is used during inference.