Zeqiang-Lai / OpenDMD

Open source implementation and models of One-step Diffusion with Distribution Matching Distillation
GNU General Public License v2.0
94 stars 11 forks source link

@Meshwa428 Hi have you solved it yet? I'm facing the same problem. I tried to move ``kwargs = dict(...)`` inside of the iteration below but it throws out a new error:I have the same problem. How did you solve it? #7

Open conquer-pan opened 1 month ago

conquer-pan commented 1 month ago
          @Meshwa428 Hi have you solved it yet? I'm facing the same problem. I tried to move ``kwargs = dict(...)`` inside of the iteration below but it throws out a new error:

RuntimeError: device >= 0 && device < num_gpus INTERNAL ASSERT FAILED at "../aten/src/ATen/cuda/CUDAContext.cpp":50, please report a bug to PyTorch. device=, num_gpus=

Originally posted by @Superviro in https://github.com/Zeqiang-Lai/OpenDMD/issues/5#issuecomment-2100107199