This commit is contained in:
comfyanonymous 2025-02-26 20:41:38 -05:00 committed by GitHub
parent 2dac065cd7
commit 50ff3a2a6a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -96,7 +96,7 @@ except:
npu_available = False
try:
import torch_mlu
import torch_mlu # noqa: F401
_ = torch.mlu.device_count()
mlu_available = torch.mlu.is_available()
except: