pyproject.toml

[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*"]