Add Func: npugraph_batch_size auto-adjust to different model

Author: chris668899

tests/singlecard/test_dynamic_npugraph_batchsize.py

@@ -17,8 +17,8 @@
 import pytest
 import torch
 
-from vllm import LLM, SamplingParams
 from torch_npu.op_plugin.atb._atb_ops import _register_atb_extensions
+from vllm import LLM, SamplingParams
 
 _register_atb_extensions()
 torch.cuda.CUDAGraph = torch.npu.NPUGraph