File size: 989 Bytes
d770a70 dd67cb7 d770a70 dd67cb7 d770a70 dd67cb7 d770a70 76bd29f d770a70 76bd29f d770a70 dd67cb7 d770a70 |
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 |
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"
[project]
name = "pulse_ecg_handler"
version = "0.1.0"
description = "PULSE ECG model handler and Cog predictor"
readme = "README.md"
requires-python = ">=3.9"
dependencies = [
# Core deep learning libraries
"torch==2.1.2",
"torchvision==0.16.2",
"transformers==4.37.2",
"tokenizers==0.15.1",
"sentencepiece==0.1.99",
"safetensors>=0.4.2",
# Accelerator and helper libraries
"accelerate==0.30.0",
"huggingface_hub==0.13.4",
# Vision and image utilities
"timm==0.6.13",
"pillow>=9.5.0",
"requests>=2.31.0",
"numpy",
"einops==0.6.1",
"einops-exts==0.0.4",
# Sentence transformer for potential extended tasks
"sentence-transformers==2.2.2",
# LLaVA multimodal model; pinned to v1.2.0
"llava @ git+https://github.com/haotian-liu/LLaVA@v1.2.0",
]
[tool.setuptools.packages.find]
exclude = ["tests*", "docs*", "examples*"] |