format

Signed-off-by: ganyi <ygan@amd.com>

Author: ganyi1996ppo

atom/plugin/vllm/model_wrapper.py

@@ -119,6 +119,7 @@ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
             model_arch = vllm_config.model_config.architectures[0]
 
         from atom.config import get_current_atom_config
+
         if self._is_draft:
             self.atom_config = get_current_atom_config()
         else:
@@ -163,7 +164,6 @@ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
         self.pp_group = get_pp_group()
         self.tp_group = get_tp_group()
 
-
     def _register_indexer_caches_with_vllm(self):
         """Register DeepseekV32IndexerCache instances with vLLM so that:
         1. vLLM discovers them via isinstance(AttentionLayerBase) for KV cache
@@ -220,8 +220,7 @@ def _register_indexer_caches_with_vllm(self):
             if prefix not in vllm_sfc:
                 vllm_sfc[prefix] = module
                 logger.info(
-                    f"Registered indexer cache in vLLM static_forward_context: "
-                    f"{prefix}"
+                    f"Registered indexer cache in vLLM static_forward_context: {prefix}"
                 )
             else:
                 logger.warning(
@@ -295,7 +294,6 @@ class ATOMMoEForCausalLM(ATOMModelBase, VllmModelForTextGeneration): ...
 class ATOMForConditionalGeneration(
     ATOMModelBase, VllmModelForTextGeneration, SupportsMultiModal, SupportsMRoPE
 ):
-
     @classmethod
     def get_placeholder_str(cls, modality: str, i: int) -> str | None:
         """