Clean up imports and move files (#14317)

This commit is contained in:
Lianmin Zheng
2025-12-02 16:31:54 -08:00
committed by GitHub
parent 5c03aa3e9d
commit ca52ed425f
28 changed files with 280 additions and 53 deletions

View File

@@ -18,6 +18,7 @@ dependencies = [
"IPython",
"aiohttp",
"anthropic>=0.20.0",
"av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' and platform_machine == 'armv7l')",
"blobfile==3.0.0",
"build",
"compressed-tensors",
@@ -38,6 +39,7 @@ dependencies = [
"ninja",
"numpy",
"nvidia-cutlass-dsl==4.2.1",
"nvidia-ml-py",
"openai-harmony==0.0.4",
"openai==2.6.1",
"orjson",
@@ -50,7 +52,6 @@ dependencies = [
"py-spy",
"pybase64",
"pydantic",
"nvidia-ml-py",
"python-multipart",
"pyzmq>=25.1.2",
"requests",
@@ -63,9 +64,8 @@ dependencies = [
"timm==1.0.16",
"torch_memory_saver==0.0.9",
"torch==2.9.1",
"torchcodec==0.7.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torchcodec does not exist in those systems. If not provided, transformer will use torchvision instead by default.
"av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' and platform_machine == 'armv7l')",
"torchaudio==2.9.1",
"torchcodec==0.7.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torchcodec does not exist in those systems. If not provided, transformer will use torchvision instead by default.
"torchvision",
"torchao==0.9.0",
"tqdm",
@@ -73,6 +73,7 @@ dependencies = [
"uvicorn",
"uvloop",
"xgrammar==0.1.27",
"grpcio==1.75.1", # keep it align with compile_proto.py
"grpcio-tools==1.75.1", # keep it align with compile_proto.py
"grpcio-reflection==1.75.1", # required by srt/entrypoints/grpc_server.py
@@ -82,17 +83,17 @@ dependencies = [
[project.optional-dependencies]
checkpoint-engine = ["checkpoint-engine==0.1.2"]
diffusion = [
"diffusers==0.35.2",
"yunchang==0.6.3.post1",
"opencv-python==4.10.0.84",
"imageio==2.36.0",
"imageio-ffmpeg==0.5.1",
"PyYAML==6.0.1",
"moviepy>=2.0.0",
"cloudpickle",
"remote-pdb",
"st_attn ==0.0.7",
"vsa==0.0.4",
"PyYAML==6.0.1",
"cloudpickle",
"diffusers==0.35.2",
"imageio==2.36.0",
"imageio-ffmpeg==0.5.1",
"moviepy>=2.0.0",
"opencv-python==4.10.0.84",
"remote-pdb",
"st_attn ==0.0.7",
"vsa==0.0.4",
"yunchang==0.6.3.post1",
]
[tool.uv.extra-build-dependencies]