File size: 1,968 Bytes
9b57ce7 |
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 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
[build-system]
requires = ["setuptools>=45", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "hpsv3"
version = "1.0.0"
description = "HPSv3: Towards Wide-Spectrum Human Preference Score - A VLM-based preference model for image quality assessment"
readme = "README.md"
requires-python = ">=3.8"
license = {text = "MIT"}
authors = [
{name = "Yunhao Shui"},
{name = "Yuhang Ma"},
]
keywords = ["machine learning", "computer vision", "human preference", "image quality", "VLM", "multimodal"]
classifiers = [
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development :: Libraries :: Python Modules",
]
dependencies = [
"torch>=2.0.0",
"torchvision>=0.15.0",
"transformers==4.45.2",
"accelerate>=0.20.0",
"datasets>=2.10.0",
"diffusers>=0.20.0",
"Pillow>=9.0.0",
"numpy>=1.20.0",
"tqdm>=4.60.0",
"pyyaml>=6.0",
"omegaconf>=2.3.0",
"opencv-python>=4.5.0",
"safetensors>=0.3.0",
"einops>=0.6.0",
"qwen-vl-utils>=0.0.8",
"timm>=0.9.0",
"deepspeed>=0.12.0",
"peft>=0.8.0",
"trl>=0.7.0",
"fire>=0.7.0"
]
[project.urls]
Homepage = "https://mizzenai.github.io/HPSv3/"
Source = "https://github.com/MizzenAI/HPSv3"
Documentation = "https://github.com/MizzenAI/HPSv3/blob/main/README.md"
Paper = "https://arxiv.org/abs/2411.07232"
[tool.setuptools.packages.find]
include = ["hpsv3*", "generate*", "evaluate*"]
[tool.setuptools.package-data]
hpsv3 = ["config/*.yaml", "config/ds_config/*.json"] |