Spaces:
Running
on
Zero
Running
on
Zero
File size: 4,432 Bytes
93b49a4 |
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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 |
import os
import subprocess
import time
from setuptools import find_packages, setup
# # # os.system('pip install --upgrade pip')
# # # os.system('python -m pip install wheel')
# # # os.system('python -m pip install urllib3==2.0')
# os.system('python -m pip install Openmim')
# os.system('python -m mim install mmengine')
# # # os.system('python -m mim install "mmcv-full==1.7.0"')
# os.system('python -m mim install "mmcv-full==1.6.2"')
# # # os.system('python -m mim install "mmcv-full"')
# # # os.system('python -m mim install "mmcv-full==1.7.2"')
# # # os.system('python -m mim install "mmcv < 2.0.0"')
# os.system('python -m mim install "mmpose==0.29.0"')
# # # os.system('python -m mim install "mmpose"')
# #
# # os.system('python -m pip install mmengine')
# # os.system('python -m pip install "mmcv-full==1.7.2"')
# # os.system('python -m pip install "mmpose==1.3.1"')
# #
# # # os.system('python -m mim install "mmcv-full==1.5.3"')
# # # os.system('python -m mim install "mmpose==0.28.1"')
# #
# # # os.system('python -m mim install "gradio==3.44.0"')
# # # os.system('python -m pip install "gradio==4.31.0"')
# os.system('python -m pip install git+https://github.com/openai/CLIP.git')
# # os.system('python -m pip install yapf==0.40.1')
# os.system('python -m pip install transformers')
# os.system('python -m mim install "gradio==3.44.0"')
def readme():
with open('README.md', encoding='utf-8') as f:
content = f.read()
return content
version_file = 'models/version.py'
def get_git_hash():
def _minimal_ext_cmd(cmd):
# construct minimal environment
env = {}
for k in ['SYSTEMROOT', 'PATH', 'HOME']:
v = os.environ.get(k)
if v is not None:
env[k] = v
# LANGUAGE is used on win32
env['LANGUAGE'] = 'C'
env['LANG'] = 'C'
env['LC_ALL'] = 'C'
out = subprocess.Popen(
cmd, stdout=subprocess.PIPE, env=env).communicate()[0]
return out
try:
out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD'])
sha = out.strip().decode('ascii')
except OSError:
sha = 'unknown'
return sha
def get_hash():
if os.path.exists('.git'):
sha = get_git_hash()[:7]
elif os.path.exists(version_file):
try:
from models.version import __version__
sha = __version__.split('+')[-1]
except ImportError:
raise ImportError('Unable to get git version')
else:
sha = 'unknown'
return sha
def write_version_py():
content = """# GENERATED VERSION FILE
# TIME: {}
__version__ = '{}'
short_version = '{}'
version_info = ({})
"""
sha = get_hash()
with open('models/VERSION', 'r') as f:
SHORT_VERSION = f.read().strip()
VERSION_INFO = ', '.join(SHORT_VERSION.split('.'))
VERSION = SHORT_VERSION + '+' + sha
version_file_str = content.format(time.asctime(), VERSION, SHORT_VERSION,
VERSION_INFO)
with open(version_file, 'w') as f:
f.write(version_file_str)
def get_version():
with open(version_file, 'r') as f:
exec(compile(f.read(), version_file, 'exec'))
return locals()['__version__']
def get_requirements(filename='requirements_setup.txt'):
here = os.path.dirname(os.path.realpath(__file__))
with open(os.path.join(here, filename), 'r') as f:
requires = [line.replace('\n', '') for line in f.readlines()]
return requires
if __name__ == '__main__':
write_version_py()
setup(
name='capex',
version=get_version(),
description='A template for pytorch projects.',
long_description=readme(),
packages=find_packages(exclude=('configs', 'tools', 'demo_text')),
package_data={'capex.ops': ['*/*.so']},
classifiers=[
'Development Status :: 4 - Beta',
'License :: OSI Approved :: Apache Software License',
'Operating System :: OS Independent',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
],
license='Apache License 2.0',
setup_requires=['pytest-runner', 'cython', 'numpy'],
tests_require=['pytest', 'xdoctest'],
install_requires=get_requirements(),
zip_safe=False)
|