File size: 404 Bytes
685ecb2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import platform
from setuptools import setup, find_packages

if platform.python_version_tuple()[:2] != ("3", "11"):
    raise RuntimeError("Python version 3.11 required")

setup(
    name="distilvit",
    version="0.1",
    packages=find_packages(),
    entry_points={
        "console_scripts": [
            "train=src.model.train:main",  # "main" is a function in "train_model.py"
        ],
    },
)