49 lines
1.5 KiB
Python
Executable File
49 lines
1.5 KiB
Python
Executable File
"""Install Compacter."""
|
|
import os
|
|
import setuptools
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
|
|
|
#os.environ['TORCH_CUDA_ARCH_LIST']="3.5;3.7;6.1;7.0;7.5;8.6+PTX"
|
|
|
|
def setup_package():
|
|
long_description = "examples_prompt"
|
|
setuptools.setup(
|
|
name='examples_prompt',
|
|
version='0.0.1',
|
|
description='textual prompt example',
|
|
long_description=long_description,
|
|
long_description_content_type='text/markdown',
|
|
author='Shengding Hu',
|
|
license='MIT License',
|
|
packages=setuptools.find_packages(
|
|
exclude=['docs', 'tests', 'scripts']),
|
|
dependency_links=[
|
|
'https://download.pytorch.org/whl/torch_stable.html',
|
|
],
|
|
classifiers=[
|
|
'Intended Audience :: Developers',
|
|
'Intended Audience :: Science/Research',
|
|
'License :: OSI Approved :: MIT License',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'Programming Language :: Python :: 3',
|
|
'Programming Language :: Python :: 3.7.10',
|
|
],
|
|
keywords='text nlp machinelearning',
|
|
# ext_modules=[
|
|
# CUDAExtension('seq2seq.projections.fwh_cuda',
|
|
# sources=[
|
|
# 'seq2seq/projections/fwh_cuda/fwh_cpp.cpp',
|
|
# 'seq2seq/projections/fwh_cuda/fwh_cu.cu',
|
|
# ]
|
|
# )
|
|
# ]
|
|
# ,
|
|
cmdclass={"build_ext": BuildExtension},
|
|
install_requires=[
|
|
],
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
setup_package()
|