RPG-DiffusionMaster
RPG-DiffusionMaster copied to clipboard
How to run it on non-default GPU or multiple GPUs?
The code can only run on the default GPU:0. Exceptions occur when device is switched to the non-default one.
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "2"
Exception has occurred: AttributeError 'str' object has no attribute 'traceback'
The above exception was the direct cause of the following exception:
File "RPG-DiffusionMaster-main/modules/errors.py", line 98, in run code() File "RPG-DiffusionMaster-main/modules/devices.py", line 76, in enable_tf32 device_id = (int(shared.cmd_opts.device_id) if shared.cmd_opts.device_id is not None and shared.cmd_opts.device_id.isdigit() else 0) or torch.cuda.current_device()
During handling of the above exception, another exception occurred:
File "RPG-DiffusionMaster-main/modules/errors.py", line 68, in display
te = traceback.TracebackException.from_exception(e)
File "RPG-DiffusionMaster-main/modules/errors.py", line 100, in run
display(task, e)
File "RPG-DiffusionMaster-main/modules/devices.py", line 84, in