setup.py

# -*- coding: utf-8 -*-
# Copyright (c) XiMing Xing. All rights reserved.
# Author: XiMing Xing

from setuptools import setup, find_packages

setup(
    name='DiffSketcher',
    packages=find_packages(),
    version='1.0.0',
    license='MIT',
    description='diffsketcher',
    author='XiMing Xing',
    author_email='ximingxing@gmail.com',
    url='https://github.com/ximinng/DiffSketcher',
    long_description_content_type='text/markdown',
    keywords=[
        'artificial intelligence',
        'aigc',
        'generative models',
        'vector graphic'
    ],
    install_requires=[
        'omegaconf',  # YAML processor
        'accelerate',  # Hugging Face - pytorch distributed configuration
        'diffusers==0.20.2',  # Hugging Face - diffusion models
        'transformers',  # Hugging Face - transformers
        'safetensors',
        'xformers',
        'einops',
        'pillow',
        'torch>=1.13.1',
        'torchvision',
        'tensorboard',
        'torchmetrics',
        'triton',
        'numba',
        'tqdm',  # progress bar
        'ftfy',
        'regex',
        'timm',  # computer vision models
        "numpy",  # numpy
        'scikit-learn',
        'scikit-fmm',
        'scipy',
        'scikit-image',
        'Pillow',  # keep the PIL.Image.Resampling deprecation away,
        'matplotlib',
        'visdom',
        'wandb',  # weights & Biases
        'opencv-python',  # cv2
        'BeautifulSoup4',
        'freetype-py',  # font
        'shapely'  # SVG
    ],
    classifiers=[
        'Development Status :: 4 - Beta',
        'Intended Audience :: Developers',
        'Topic :: Scientific/Engineering :: Artificial Intelligence',
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python :: 3.10',
    ],
)