File size: 1,018 Bytes
4357d9b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | #!/usr/bin/env python3
from __future__ import annotations
import argparse
import importlib.util
import sys
DEP_GROUPS = {
"diffsynth": [
"torch",
"transformers",
"diffsynth",
"safetensors",
"imageio",
"PIL",
"requests",
"vgen",
],
"qwen35_27b": [
"torch",
"transformers",
"vllm",
],
"cosyvoice": [
"torch",
"torchaudio",
"cosyvoice",
],
}
def main() -> None:
parser = argparse.ArgumentParser(description="Check required Python imports.")
parser.add_argument("--group", required=True, choices=sorted(DEP_GROUPS))
args = parser.parse_args()
missing = [name for name in DEP_GROUPS[args.group] if importlib.util.find_spec(name) is None]
if missing:
print(f"[FAIL] {args.group}: missing {', '.join(missing)}", file=sys.stderr)
raise SystemExit(1)
print(f"[ OK ] {args.group}: all imports available")
if __name__ == "__main__":
main()
|