forked from lsdefine/attention-is-all-you-need-keras
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
executable file
·27 lines (23 loc) · 984 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
import setuptools
from setuptools.extension import Extension
import numpy as np
with open('requirements.txt') as f:
requirements = f.read().splitlines()
setuptools.setup(
name = 'keras-transformer',
version = '0.1.0',
description = 'Keras implementation of the Transformer Network for sequence to sequence translation. '
'This is a modification of https://github.com/Lsdefine/attention-is-all-you-need-keras/',
url = 'https://github.com/code-lava/keras-transformer/',
author = 'Fares Abawi',
author_email = 'fares.abawi@informatik.uni-hamburg.de',
maintainer = 'Fares Abawi',
maintainer_email = 'fares.abawi@informatik.uni-hamburg.de',
packages = setuptools.find_packages(),
install_requires = requirements,
entry_points = {
'console_scripts': [
'transformer-train=keras_transformer.bin.train:main'
],
}
)