gaotingquan před 9 měsíci
rodič
revize
64712bdb93
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  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):