| [project] |
| name = "pool" |
| version = "0.1.0" |
| description = "Add your description here" |
| readme = "README.md" |
| requires-python = ">=3.12" |
| dependencies = [ |
| "accelerate>=1.13.0", |
| "bitsandbytes>=0.49.2; sys_platform == 'linux' and platform_machine == 'x86_64'", |
| "chromadb>=0.5.0", |
| "datasets>=4.8.4", |
| "fastapi>=0.135.2", |
| "gradio>=4.0.0", |
| "huggingface-hub>=1.8.0", |
| "peft>=0.18.1", |
| "redis>=5.0.0", |
| "pymongo>=4.6.0", |
| "motor>=3.4.0", |
| "minio>=7.2.0", |
| "python-multipart>=0.0.9", |
| "sentence-transformers>=3.0.0", |
| "torch>=2.5.0", |
| "torchaudio>=2.5.0", |
| "transformers>=5.4.0", |
| "trl>=0.29.1", |
| "uvicorn>=0.42.0", |
| "torchvision>=0.20.0", |
| "wrapt>=2.1.2", |
| ] |
|
|
| [tool.uv] |
| |
| |
| |
| environments = [ |
| "sys_platform == 'darwin' and platform_machine == 'arm64'", |
| "sys_platform == 'linux' and platform_machine == 'x86_64'", |
| ] |
|
|
| [tool.uv.sources] |
| |
| torch = [{ index = "pytorch-cu121", marker = "sys_platform == 'linux'" }] |
| torchvision = [{ index = "pytorch-cu121", marker = "sys_platform == 'linux'" }] |
| torchaudio = [{ index = "pytorch-cu121", marker = "sys_platform == 'linux'" }] |
|
|
| [[tool.uv.index]] |
| name = "pytorch-cu121" |
| url = "https://download.pytorch.org/whl/cu121" |
| explicit = true |
|
|
