BlackPuuuudding

Results 3 issues of BlackPuuuudding

After running the `sample.py` file, the `prompts.json` file is not generated. How can I generate this file?

Why does my training run out of memory when the batch size is set to 4, and out of memory when the batch size is set to 2 during multi-GPU...