style: remove unused import (#184)

lightx2v/api_multi_servers.py

@@ -5,9 +5,8 @@ import socket
 import subprocess
 import time
 from dataclasses import dataclass
-from typing import Dict, List, Optional
+from typing import Optional
 
-import psutil
 import requests
 from loguru import logger
 

lightx2v/api_server.py

@@ -9,7 +9,6 @@ from loguru import logger
 
 from lightx2v.server.api import ApiServer
 from lightx2v.server.service import DistributedInferenceService
-from lightx2v.server.utils import ProcessManager
 
 
 def create_signal_handler(inference_service: DistributedInferenceService):

lightx2v/common/apis/dit.py

 import argparse
 import json
-import os
-from typing import Optional
 
-import torch
 import uvicorn
 from fastapi import FastAPI
 from loguru import logger
 from pydantic import BaseModel
 
 from lightx2v.common.ops import *
-from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner
-from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner
-from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner
-from lightx2v.models.runners.wan.wan_runner import WanRunner
-from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner
+from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_runner import WanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner  # noqa: F401
 from lightx2v.utils.profiler import ProfilingContext
 from lightx2v.utils.registry_factory import RUNNER_REGISTER
 from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter

lightx2v/common/apis/image_encoder.py

 import argparse
 import json
-import os
 
-import torch
-import torchvision.transforms.functional as TF
 import uvicorn
 from fastapi import FastAPI
 from loguru import logger
 from pydantic import BaseModel
 
-from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner
-from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner
-from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner
-from lightx2v.models.runners.wan.wan_runner import WanRunner
-from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner
+from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_runner import WanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner  # noqa: F401
 from lightx2v.utils.profiler import ProfilingContext
 from lightx2v.utils.registry_factory import RUNNER_REGISTER
 from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter

lightx2v/common/apis/prompt_enhancer.py

 import argparse
-import json
-from typing import Optional
 
 import uvicorn
 from fastapi import FastAPI

lightx2v/common/apis/text_encoder.py

 import argparse
 import json
-import os
 from typing import Optional
 
-import torch
 import uvicorn
 from fastapi import FastAPI
 from loguru import logger
 from pydantic import BaseModel
 
-from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner
-from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner
-from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner
-from lightx2v.models.runners.wan.wan_runner import WanRunner
-from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner
+from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_runner import WanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner  # noqa: F401
 from lightx2v.utils.profiler import ProfilingContext
 from lightx2v.utils.registry_factory import RUNNER_REGISTER
 from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter

lightx2v/common/apis/vae.py

 import argparse
 import json
-import os
 from typing import Optional
 
-import numpy as np
-import torch
-import torchvision
-import torchvision.transforms.functional as TF
 import uvicorn
 from fastapi import FastAPI
 from loguru import logger
 from pydantic import BaseModel
 
 from lightx2v.common.ops import *
-from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner
-from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner
-from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner
-from lightx2v.models.runners.wan.wan_runner import WanRunner
-from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner
+from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_runner import WanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner  # noqa: F401
 from lightx2v.utils.profiler import ProfilingContext
 from lightx2v.utils.registry_factory import RUNNER_REGISTER
 from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter

lightx2v/common/ops/attn/flash_attn.py

 from loguru import logger
 
 try:
-    import flash_attn
+    import flash_attn  # noqa: F401
     from flash_attn.flash_attn_interface import flash_attn_varlen_func
 except ImportError:
     logger.info("flash_attn_varlen_func not found, please install flash_attn2 first")

lightx2v/common/ops/attn/sparge_attn.py

-import torch
 import torch.nn as nn
 from loguru import logger
 

lightx2v/common/ops/attn/utils/ring_comm.py

@@ -2,7 +2,6 @@ from typing import Optional
 
 import torch
 import torch.distributed as dist
-from loguru import logger
 
 
 class RingComm:

lightx2v/common/ops/mm/mm_weight_calib.py

 import torch
 
-from lightx2v.utils.quant_utils import FloatQuantizer, IntegerQuantizer
+from lightx2v.utils.quant_utils import FloatQuantizer
 from lightx2v.utils.registry_factory import MM_WEIGHT_REGISTER
 
 from .mm_weight import MMWeight

lightx2v/common/transformer_infer/transformer_infer.py

 import math
 from abc import ABC, abstractmethod
 
-import torch
-
 
 class BaseTransformerInfer(ABC):
     @abstractmethod

lightx2v/infer.py

 import argparse
-import json
 
-import torch
 import torch.distributed as dist
 from loguru import logger
 
 from lightx2v.common.ops import *
-from lightx2v.models.runners.cogvideox.cogvidex_runner import CogvideoxRunner
+from lightx2v.models.runners.cogvideox.cogvidex_runner import CogvideoxRunner  # noqa: F401
 from lightx2v.models.runners.graph_runner import GraphRunner
-from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner
-from lightx2v.models.runners.wan.wan_audio_runner import Wan22MoeAudioRunner, WanAudioRunner
-from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner
-from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner
-from lightx2v.models.runners.wan.wan_runner import Wan22MoeRunner, WanRunner
-from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner
+from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_audio_runner import Wan22MoeAudioRunner, WanAudioRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_runner import Wan22MoeRunner, WanRunner  # noqa: F401
+from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner  # noqa: F401
 from lightx2v.utils.envs import *
 from lightx2v.utils.profiler import ProfilingContext
 from lightx2v.utils.registry_factory import RUNNER_REGISTER

lightx2v/models/input_encoders/hf/llava/model.py

-import numpy as np
 import torch
-import torchvision.transforms as transforms
-from PIL import Image
-from loguru import logger
 from transformers import AutoTokenizer, CLIPImageProcessor, LlavaForConditionalGeneration
 
 

lightx2v/models/input_encoders/hf/t5/model.py

 # Modified from transformers.models.t5.modeling_t5
 # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
-import logging
 import math
 import os
 

lightx2v/models/input_encoders/hf/xlm_roberta/model.py

 # Modified from ``https://github.com/openai/CLIP'' and ``https://github.com/mlfoundations/open_clip''
 # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
-import logging
 import math
 
 import torch

lightx2v/models/networks/hunyuan/infer/utils_bf16.py

-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import Tuple, Union
 
 import torch
 

lightx2v/models/networks/hunyuan/infer/utils_fp32.py

-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import Tuple, Union
 
 import torch
 

lightx2v/models/networks/wan/audio_model.py

 import glob
 import os
-import time
 
 import torch
-from safetensors import safe_open
 
 from lightx2v.common.ops.attn.radial_attn import MaskMap
 from lightx2v.models.networks.wan.infer.audio.post_wan_audio_infer import WanAudioPostInfer
 from lightx2v.models.networks.wan.infer.audio.pre_wan_audio_infer import WanAudioPreInfer
-from lightx2v.models.networks.wan.infer.feature_caching.transformer_infer import WanTransformerInferTeaCaching
 from lightx2v.models.networks.wan.model import WanModel
 from lightx2v.models.networks.wan.weights.post_weights import WanPostWeights
 from lightx2v.models.networks.wan.weights.pre_weights import WanPreWeights

lightx2v/models/networks/wan/distill_model.py

-import glob
-import json
 import os
-import sys
 
 import torch
 from loguru import logger
-from safetensors import safe_open
 
 from lightx2v.models.networks.wan.model import WanModel
 from lightx2v.models.networks.wan.weights.post_weights import WanPostWeights