diff --git a/deepmd/jax/infer/deep_eval.py b/deepmd/jax/infer/deep_eval.py index 3a9cead9d8..76f044a327 100644 --- a/deepmd/jax/infer/deep_eval.py +++ b/deepmd/jax/infer/deep_eval.py @@ -68,7 +68,7 @@ class DeepEval(DeepEvalBackend): The output definition of the model. *args : list Positional arguments. - auto_batch_size : bool or int or AutomaticBatchSize, default: False + auto_batch_size : bool or int or AutoBatchSize, default: True If True, automatic batch size will be used. If int, it will be used as the initial batch size. neighbor_list : ase.neighborlist.NewPrimitiveNeighborList, optional