style: remove unused import (#184)

lightx2v/models/video_encoders/hf/wan/vae.py

@@ -2,7 +2,6 @@
 import logging
 
 import torch
-import torch.cuda.amp as amp
 import torch.distributed as dist
 import torch.nn as nn
 import torch.nn.functional as F

lightx2v/models/video_encoders/hf/wan/vae_2_2.py

@@ -2,7 +2,6 @@
 import logging
 
 import torch
-import torch.cuda.amp as amp
 import torch.nn as nn
 import torch.nn.functional as F
 from einops import rearrange

lightx2v/utils/prompt_enhancer.py

@@ -4,7 +4,7 @@ import torch
 from loguru import logger
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-from lightx2v.utils.profiler import ProfilingContext, ProfilingContext4Debug
+from lightx2v.utils.profiler import ProfilingContext
 
 sys_prompt = """
 Transform the short prompt into a detailed video-generation caption using this structure:

lightx2v/utils/service_utils.py

@@ -4,7 +4,7 @@ import signal
 import sys
 import threading
 from datetime import datetime
-from typing import List, Optional
+from typing import Optional
 
 import psutil
 import torch

pyproject.toml

@@ -19,7 +19,7 @@ indent-width = 4
 
 
 [tool.ruff.lint]
-extend-select = ["I"]
+extend-select = ["I", "F401"]
 ignore = ["F"]
 
 [tool.ruff.lint.per-file-ignores]

scripts/server/post_multi_servers.py

-import random
-import string
 import threading
 import time
-from datetime import datetime
 
 import requests
 from loguru import logger

tools/extract/lora_extractor.py

@@ -7,8 +7,7 @@ Extract LoRA weights from the difference between two models
 
 import argparse
 import os
-from pathlib import Path
-from typing import Any, Dict, Optional
+from typing import Dict, Optional
 
 import torch
 from safetensors import safe_open

tools/extract/lora_merger.py

@@ -7,8 +7,7 @@ Merge a source model with LoRA weights to create a new model
 
 import argparse
 import os
-from pathlib import Path
-from typing import Any, Dict, Optional
+from typing import Dict, Optional
 
 import torch
 from safetensors import safe_open