Files
mask-ddpm/example/test_gpu.py