-
Notifications
You must be signed in to change notification settings - Fork 5
/
setup.py
28 lines (27 loc) · 824 Bytes
/
setup.py
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
from setuptools import setup, find_packages
setup(
name="ohara",
version="0.1.3",
packages=find_packages(),
author="joey00072",
author_email="[email protected]",
description="A collection of implementations of LLM, papers, and other models.",
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
url="https://github.com/joey00072/ohara",
install_requires=[
# Add your project dependencies here
"lightning>=2.3.0",
"transformers>=4.46.0",
"datasets>=3.1.0",
"wandb>=0.18.0",
"rich>=13",
"tensorboard>=2.18.0",
],
entry_points={
"console_scripts": [
"ohara=ohara.main:main", # Adjust 'ohara.main:main' as necessary
],
},
python_requires=">=3.6",
)