gaotingquan пре 9 месеци
родитељ
комит
64712bdb93
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      paddlex/utils/device.py

+ 1 - 1
paddlex/utils/device.py

@@ -26,7 +26,7 @@ from .custom_device_whitelist import (
     GCU_WHITELIST,
 )
 
-SUPPORTED_DEVICE_TYPE = ["cpu", "gpu", "xpu", "npu", "mlu", "gcu"]
+SUPPORTED_DEVICE_TYPE = ["cpu", "gpu", "xpu", "npu", "mlu", "gcu", "dcu"]
 
 
 def _constr_device(device_type, device_ids):