set auto_batch_size to True

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jan 24, 2024 · 60e44b4 · 60e44b4
1 parent c2c9e72
commit 60e44b4
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/deepmd_utils/infer/model_devi.py b/deepmd_utils/infer/model_devi.py
@@ -13,9 +13,6 @@
 from deepmd_utils.infer.deep_pot import (
     DeepPot,
 )
-from deepmd_utils.utils.batch_size import (
-    AutoBatchSize,
-)
 from deepmd_utils.utils.data import (
     DeepmdData,
 )
@@ -395,9 +392,8 @@ def make_model_devi(
     **kwargs
         Arbitrary keyword arguments.
     """
-    auto_batch_size = AutoBatchSize()
     # init models
-    dp_models = [DeepPot(model, auto_batch_size=auto_batch_size) for model in models]
+    dp_models = [DeepPot(model, auto_batch_size=True) for model in models]
 
     # check type maps
     tmaps = [dp.get_type_map() for dp in dp_models]