use torch_directml will ModuleNotFoundError: No module named 'torch.privateuseone'

#259
by cofmass - opened

Traceback (most recent call last):
File "C:\ClothesRoomDjangoBackend\OOTDiffusion\run\run_ootd.py", line 95, in
main()
File "C:\ClothesRoomDjangoBackend\OOTDiffusion\run\run_ootd.py", line 35, in main
model = OOTDiffusionHD(device=device)
File "C:\ClothesRoomDjangoBackend\OOTDiffusion\ootd\inference_ootd_hd.py", line 26, in init
).to(self.device)
File "C:\Users\96307.conda\envs\django4\lib\site-packages\diffusers\pipelines\pipeline_utils.py", line 864, in to
module.to(device, dtype)
File "C:\Users\96307.conda\envs\django4\lib\site-packages\torch\nn\modules\module.py", line 1174, in to
return self._apply(convert)
File "C:\Users\96307.conda\envs\django4\lib\site-packages\torch\nn\modules\module.py", line 780, in _apply
module._apply(fn)
File "C:\Users\96307.conda\envs\django4\lib\site-packages\torch\nn\modules\module.py", line 780, in _apply
module._apply(fn)
File "C:\Users\96307.conda\envs\django4\lib\site-packages\torch\nn\modules\module.py", line 805, in _apply
param_applied = fn(param)
File "C:\Users\96307.conda\envs\django4\lib\site-packages\torch\nn\modules\module.py", line 1160, in convert
return t.to(
ModuleNotFoundError: No module named 'torch.privateuseone'
if use torch_directml will make this error

Your need to confirm your account before you can post a new comment.

Sign up or log in to comment