[BUG] Linear model for pytorch backend

[ChiahsinChu]

Bug summary

I try to run the example of D3 correction with pytorch backend. However, it shows an error of "unknown model type". The running command and the output error are shown below.

DeePMD-kit Version

DeePMD-kit v3.0.0b4 (commit: 0b3f860)

Backend and its version

PyTorch v2.2.1-g6c8c5ad5eaf

How did you download the software?

Built from source

Input Files, Running Commands, Error Log, etc.

Running command:

cd $deepmd_dir/examples/water/d3
dp --pt train input.json 

where $deepmd_dir is the path to deepmd source code.

Error log:

Traceback (most recent call last):
  File "/home/jxzhu/apps/miniconda3/envs/deepmd-latest/bin/dp", line 8, in <module>
    sys.exit(main())
  File "/home/jxzhu/apps/deepmd/official_devel/deepmd/main.py", line 929, in main
    deepmd_main(args)
  File "/home/jxzhu/apps/miniconda3/envs/deepmd-latest/lib/python3.10/site-packages/torch/distributed/elastic/multiprocessing/errors/__init__.py", line 347, in wrapper
    return f(*args, **kwargs)
  File "/home/jxzhu/apps/deepmd/official_devel/deepmd/pt/entrypoints/main.py", line 503, in main
    train(FLAGS)
  File "/home/jxzhu/apps/deepmd/official_devel/deepmd/pt/entrypoints/main.py", line 305, in train
    config["model"], min_nbor_dist = BaseModel.update_sel(
  File "/home/jxzhu/apps/deepmd/official_devel/deepmd/dpmodel/model/base_model.py", line 194, in update_sel
    cls = cls.get_class_by_type(model_type)
  File "/home/jxzhu/apps/deepmd/official_devel/deepmd/utils/plugin.py", line 154, in get_class_by_type
    raise RuntimeError(f"Unknown {name} type: {class_type}. {dym_message}")
RuntimeError: Unknown model type: linear_ener. 

Steps to Reproduce

cd $deepmd_dir/examples/water/d3
dp --pt train input.json 

Further Information, Files, and Links

No response