Lint

Author: jackzhxng

backends/arm/test/models/test_llama.py

@@ -22,13 +22,13 @@
     TosaPipelineBI,
     TosaPipelineMI,
 )
-
-from executorch.extension.llm.export.config.llm_config import LlmConfig
 from executorch.examples.models.llama.export_llama_lib import (
     build_args_parser,
     get_llama_model,
 )
 
+from executorch.extension.llm.export.config.llm_config import LlmConfig
+
 input_t = Tuple[torch.Tensor]
 
 # Add project dir to sys path to workaround importlib.import_module() conditions in model_factory.py

examples/apple/mps/scripts/mps_example.py

@@ -19,8 +19,6 @@
 from executorch.devtools.bundled_program.serialize import (
     serialize_from_bundled_program_to_flatbuffer,
 )
-
-from executorch.extension.llm.export.config.llm_config import LlmConfig
 from executorch.exir import (
     EdgeCompileConfig,
     EdgeProgramManager,
@@ -31,6 +29,8 @@
 from executorch.exir.capture._config import ExecutorchBackendConfig
 from executorch.extension.export_util.utils import export_to_edge, save_pte_program
 
+from executorch.extension.llm.export.config.llm_config import LlmConfig
+
 from ....models import MODEL_NAME_TO_MODEL
 from ....models.model_factory import EagerModelFactory
 

examples/models/llama/export_llama_lib.py

@@ -27,15 +27,15 @@
 from executorch.devtools.backend_debug import print_delegation_info
 
 from executorch.devtools.etrecord import generate_etrecord as generate_etrecord_func
-
-from executorch.extension.llm.export.config.llm_config import LlmConfig
 from executorch.examples.models.llama.hf_download import (
     download_and_convert_hf_checkpoint,
 )
 from executorch.exir.passes.init_mutable_pass import InitializedMutableBufferPass
 
 from executorch.extension.llm.export.builder import DType, LLMEdgeManager
 
+from executorch.extension.llm.export.config.llm_config import LlmConfig
+
 from executorch.extension.llm.export.partitioner_lib import (
     get_coreml_partitioner,
     get_mps_partitioner,

examples/models/llama/model.py

@@ -15,11 +15,11 @@
     get_checkpoint_dtype,
     get_default_model_resource_dir,
 )
-
-from executorch.extension.llm.export.config.llm_config import LlmConfig
 from executorch.examples.models.llama.llama_transformer import construct_transformer
 from executorch.examples.models.llama.model_args import ModelArgs
 from executorch.examples.models.llama.rope import Rope
+
+from executorch.extension.llm.export.config.llm_config import LlmConfig
 from torchao.utils import TorchAOBaseTensor
 
 try:

examples/models/llama/runner/eager.py

@@ -9,15 +9,15 @@
 from typing import Optional, Type
 
 import torch
-
-from executorch.extension.llm.export.config.llm_config import LlmConfig
 from executorch.examples.models.llama.export_llama_lib import (
     _prepare_for_llama_export,
     build_args_parser as _build_args_parser,
 )
 from executorch.examples.models.llama.runner.generation import LlamaRunner
 from executorch.extension.llm.export.builder import LLMEdgeManager
 
+from executorch.extension.llm.export.config.llm_config import LlmConfig
+
 
 class EagerLlamaRunner(LlamaRunner):
     """

examples/models/llama/tests/test_export_llama_lib.py

@@ -7,11 +7,11 @@
 import unittest
 
 from executorch.devtools.backend_debug import get_delegation_info
-from executorch.extension.llm.export.config.llm_config import LlmConfig
 from executorch.examples.models.llama.export_llama_lib import (
     _export_llama,
     build_args_parser,
 )
+from executorch.extension.llm.export.config.llm_config import LlmConfig
 
 UNWANTED_OPS = [
     "aten_permute_copy_default",

examples/models/llama3_2_vision/runner/eager.py

@@ -8,14 +8,14 @@
 from typing import Optional
 
 import torch
-from executorch.extension.llm.export.config.llm_config import LlmConfig
 
 from executorch.examples.models.llama.export_llama_lib import _prepare_for_llama_export
 from executorch.examples.models.llama.runner.eager import execute_runner
 from executorch.examples.models.llama3_2_vision.runner.generation import (
     TorchTuneLlamaRunner,
 )
 from executorch.extension.llm.export import LLMEdgeManager
+from executorch.extension.llm.export.config.llm_config import LlmConfig
 
 
 class EagerLlamaRunner(TorchTuneLlamaRunner):

examples/models/llava/export_llava.py

@@ -16,7 +16,6 @@
     get_symmetric_quantization_config,
     XNNPACKQuantizer,
 )
-from executorch.extension.llm.export.config.llm_config import LlmConfig
 from executorch.examples.models.llama.export_llama_lib import (
     get_quantizer_and_quant_params,
 )
@@ -44,6 +43,7 @@
     HintBasedSymShapeEvalPass,
 )
 from executorch.extension.llm.export.builder import DType, LLMEdgeManager
+from executorch.extension.llm.export.config.llm_config import LlmConfig
 from executorch.util.activation_memory_profiler import generate_memory_trace
 from pytorch_tokenizers.llama2c import Llama2cTokenizer as Tokenizer
 from torch.export import Dim

extension/llm/export/export_llm.py

@@ -34,9 +34,9 @@
 from typing import Any, List, Tuple
 
 import hydra
+from executorch.examples.models.llama.export_llama_lib import export_llama
 
 from executorch.extension.llm.export.config.llm_config import LlmConfig
-from executorch.examples.models.llama.export_llama_lib import export_llama
 from hydra.core.config_store import ConfigStore
 from omegaconf import OmegaConf