Skip to content

GitLab

Commit c0b36010 authored by PengGao's avatar PengGao Committed by GitHub
Browse files

style: remove unused import (#184)

parent f6d4fc85
Hide whitespace changes
Inline Side-by-side
Showing with 32 additions and 68 deletions
lightx2v/api_multi_servers.py
View file @ c0b36010
...@@ -5,9 +5,8 @@ import socket ...@@ -5,9 +5,8 @@ import socket
import subprocess import subprocess
import time import time
from dataclasses import dataclass from dataclasses import dataclass
from typing import Dict, List, Optional from typing import Optional
import psutil
import requests import requests
from loguru import logger from loguru import logger
......
lightx2v/api_server.py
View file @ c0b36010
...@@ -9,7 +9,6 @@ from loguru import logger ...@@ -9,7 +9,6 @@ from loguru import logger
from lightx2v.server.api import ApiServer from lightx2v.server.api import ApiServer
from lightx2v.server.service import DistributedInferenceService from lightx2v.server.service import DistributedInferenceService
from lightx2v.server.utils import ProcessManager
def create_signal_handler(inference_service: DistributedInferenceService): def create_signal_handler(inference_service: DistributedInferenceService):
......
lightx2v/common/apis/dit.py
View file @ c0b36010
import argparse import argparse
import json import json
import os
from typing import Optional
import torch
import uvicorn import uvicorn
from fastapi import FastAPI from fastapi import FastAPI
from loguru import logger from loguru import logger
from pydantic import BaseModel from pydantic import BaseModel
from lightx2v.common.ops import * from lightx2v.common.ops import *
from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner # noqa: F401
from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner # noqa: F401
from lightx2v.models.runners.wan.wan_runner import WanRunner from lightx2v.models.runners.wan.wan_runner import WanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner # noqa: F401
from lightx2v.utils.profiler import ProfilingContext from lightx2v.utils.profiler import ProfilingContext
from lightx2v.utils.registry_factory import RUNNER_REGISTER from lightx2v.utils.registry_factory import RUNNER_REGISTER
from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter
......
lightx2v/common/apis/image_encoder.py
View file @ c0b36010
import argparse import argparse
import json import json
import os
import torch
import torchvision.transforms.functional as TF
import uvicorn import uvicorn
from fastapi import FastAPI from fastapi import FastAPI
from loguru import logger from loguru import logger
from pydantic import BaseModel from pydantic import BaseModel
from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner # noqa: F401
from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner # noqa: F401
from lightx2v.models.runners.wan.wan_runner import WanRunner from lightx2v.models.runners.wan.wan_runner import WanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner # noqa: F401
from lightx2v.utils.profiler import ProfilingContext from lightx2v.utils.profiler import ProfilingContext
from lightx2v.utils.registry_factory import RUNNER_REGISTER from lightx2v.utils.registry_factory import RUNNER_REGISTER
from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter
......
lightx2v/common/apis/prompt_enhancer.py
View file @ c0b36010
import argparse import argparse
import json
from typing import Optional
import uvicorn import uvicorn
from fastapi import FastAPI from fastapi import FastAPI
......
lightx2v/common/apis/text_encoder.py
View file @ c0b36010
import argparse import argparse
import json import json
import os
from typing import Optional from typing import Optional
import torch
import uvicorn import uvicorn
from fastapi import FastAPI from fastapi import FastAPI
from loguru import logger from loguru import logger
from pydantic import BaseModel from pydantic import BaseModel
from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner # noqa: F401
from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner # noqa: F401
from lightx2v.models.runners.wan.wan_runner import WanRunner from lightx2v.models.runners.wan.wan_runner import WanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner # noqa: F401
from lightx2v.utils.profiler import ProfilingContext from lightx2v.utils.profiler import ProfilingContext
from lightx2v.utils.registry_factory import RUNNER_REGISTER from lightx2v.utils.registry_factory import RUNNER_REGISTER
from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter
......
lightx2v/common/apis/vae.py
View file @ c0b36010
import argparse import argparse
import json import json
import os
from typing import Optional from typing import Optional
import numpy as np
import torch
import torchvision
import torchvision.transforms.functional as TF
import uvicorn import uvicorn
from fastapi import FastAPI from fastapi import FastAPI
from loguru import logger from loguru import logger
from pydantic import BaseModel from pydantic import BaseModel
from lightx2v.common.ops import * from lightx2v.common.ops import *
from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner # noqa: F401
from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner # noqa: F401
from lightx2v.models.runners.wan.wan_runner import WanRunner from lightx2v.models.runners.wan.wan_runner import WanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner # noqa: F401
from lightx2v.utils.profiler import ProfilingContext from lightx2v.utils.profiler import ProfilingContext
from lightx2v.utils.registry_factory import RUNNER_REGISTER from lightx2v.utils.registry_factory import RUNNER_REGISTER
from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter from lightx2v.utils.service_utils import BaseServiceStatus, ImageTransporter, ProcessManager, TaskStatusMessage, TensorTransporter
......
lightx2v/common/ops/attn/flash_attn.py
View file @ c0b36010
from loguru import logger from loguru import logger
try: try:
import flash_attn import flash_attn # noqa: F401
from flash_attn.flash_attn_interface import flash_attn_varlen_func from flash_attn.flash_attn_interface import flash_attn_varlen_func
except ImportError: except ImportError:
logger.info("flash_attn_varlen_func not found, please install flash_attn2 first") logger.info("flash_attn_varlen_func not found, please install flash_attn2 first")
......
lightx2v/common/ops/attn/sparge_attn.py
View file @ c0b36010
import torch
import torch.nn as nn import torch.nn as nn
from loguru import logger from loguru import logger
......
lightx2v/common/ops/attn/utils/ring_comm.py
View file @ c0b36010
...@@ -2,7 +2,6 @@ from typing import Optional ...@@ -2,7 +2,6 @@ from typing import Optional
import torch import torch
import torch.distributed as dist import torch.distributed as dist
from loguru import logger
class RingComm: class RingComm:
......
lightx2v/common/ops/mm/mm_weight_calib.py
View file @ c0b36010
import torch 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 lightx2v.utils.registry_factory import MM_WEIGHT_REGISTER
from .mm_weight import MMWeight from .mm_weight import MMWeight
......
lightx2v/common/transformer_infer/transformer_infer.py
View file @ c0b36010
import math import math
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
import torch
class BaseTransformerInfer(ABC): class BaseTransformerInfer(ABC):
@abstractmethod @abstractmethod
......
lightx2v/infer.py
View file @ c0b36010
import argparse import argparse
import json
import torch
import torch.distributed as dist import torch.distributed as dist
from loguru import logger from loguru import logger
from lightx2v.common.ops import * 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.graph_runner import GraphRunner
from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner from lightx2v.models.runners.hunyuan.hunyuan_runner import HunyuanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_audio_runner import Wan22MoeAudioRunner, WanAudioRunner from lightx2v.models.runners.wan.wan_audio_runner import Wan22MoeAudioRunner, WanAudioRunner # noqa: F401
from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner from lightx2v.models.runners.wan.wan_causvid_runner import WanCausVidRunner # noqa: F401
from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner from lightx2v.models.runners.wan.wan_distill_runner import WanDistillRunner # noqa: F401
from lightx2v.models.runners.wan.wan_runner import Wan22MoeRunner, WanRunner from lightx2v.models.runners.wan.wan_runner import Wan22MoeRunner, WanRunner # noqa: F401
from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner from lightx2v.models.runners.wan.wan_skyreels_v2_df_runner import WanSkyreelsV2DFRunner # noqa: F401
from lightx2v.utils.envs import * from lightx2v.utils.envs import *
from lightx2v.utils.profiler import ProfilingContext from lightx2v.utils.profiler import ProfilingContext
from lightx2v.utils.registry_factory import RUNNER_REGISTER from lightx2v.utils.registry_factory import RUNNER_REGISTER
......
lightx2v/models/input_encoders/hf/llava/model.py
View file @ c0b36010
import numpy as np
import torch import torch
import torchvision.transforms as transforms
from PIL import Image
from loguru import logger
from transformers import AutoTokenizer, CLIPImageProcessor, LlavaForConditionalGeneration from transformers import AutoTokenizer, CLIPImageProcessor, LlavaForConditionalGeneration
......
lightx2v/models/input_encoders/hf/t5/model.py
View file @ c0b36010
# Modified from transformers.models.t5.modeling_t5 # Modified from transformers.models.t5.modeling_t5
# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved. # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
import logging
import math import math
import os import os
......
lightx2v/models/input_encoders/hf/xlm_roberta/model.py
View file @ c0b36010
# Modified from ``https://github.com/openai/CLIP'' and ``https://github.com/mlfoundations/open_clip'' # 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. # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
import logging
import math import math
import torch import torch
......
lightx2v/models/networks/hunyuan/infer/utils_bf16.py
View file @ c0b36010
from typing import Any, Dict, List, Optional, Tuple, Union from typing import Tuple, Union
import torch import torch
......
lightx2v/models/networks/hunyuan/infer/utils_fp32.py
View file @ c0b36010
from typing import Any, Dict, List, Optional, Tuple, Union from typing import Tuple, Union
import torch import torch
......
lightx2v/models/networks/wan/audio_model.py
View file @ c0b36010
import glob import glob
import os import os
import time
import torch import torch
from safetensors import safe_open
from lightx2v.common.ops.attn.radial_attn import MaskMap 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.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.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.model import WanModel
from lightx2v.models.networks.wan.weights.post_weights import WanPostWeights from lightx2v.models.networks.wan.weights.post_weights import WanPostWeights
from lightx2v.models.networks.wan.weights.pre_weights import WanPreWeights from lightx2v.models.networks.wan.weights.pre_weights import WanPreWeights
......
lightx2v/models/networks/wan/distill_model.py
View file @ c0b36010
import glob
import json
import os import os
import sys
import torch import torch
from loguru import logger from loguru import logger
from safetensors import safe_open
from lightx2v.models.networks.wan.model import WanModel from lightx2v.models.networks.wan.model import WanModel
from lightx2v.models.networks.wan.weights.post_weights import WanPostWeights from lightx2v.models.networks.wan.weights.post_weights import WanPostWeights
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment