Hello authors, Great work! I added a quick PR to adapt generation to run on fewer than 8 GPUs if needed - https://github.com/uclaml/SPPO/pull/24. This is a minimally invasive change
