28 lines
793 B
Python
28 lines
793 B
Python
from setuptools import setup, find_packages
|
|
|
|
setup(
|
|
name='chatbug',
|
|
version='0.1.0',
|
|
description='A conversational AI chatbot',
|
|
author='Florin Tobler',
|
|
author_email='florin.tobler@hotmail.com',
|
|
packages=find_packages(exclude=["tests"]),
|
|
install_requires=[
|
|
'transformers',
|
|
'accelerate',
|
|
'bitsandbytes',
|
|
'pytest',
|
|
'pywebview',
|
|
],
|
|
entry_points={
|
|
'console_scripts': [
|
|
'chatbug=chatbug.llama:main_func',
|
|
# a^ b^ c^ d^
|
|
# a => the command line argument
|
|
# b => the package name
|
|
# c => the file name in the package (same as imports)
|
|
# d => the function to call
|
|
'chatbugui=chatbug.ui.__main__:start_ui',
|
|
],
|
|
},
|
|
) |