huangyp/MuseV
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update setup and gradio (#64)

Browse Source 
* update requirements, setup, support pip install
This commit is contained in:
xzqjack 2024-04-20 14:59:30 +08:00 committed by GitHub
parent d3534cfe56
commit 5b81f027c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 40 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
requirements.txt
View File

@ -1,12 +1,16 @@
diffusers @ git+https://github.com/TMElyralab/diffusers.git@tme
ip_adapter @ git+https://github.com/tencent-ailab/IP-Adapter.git@main
clip @ git+https://github.com/openai/CLIP.git@main
controlnet_aux @ git+https://github.com/TMElyralab/controlnet_aux.git@tme
mmcm @ git+https://github.com/TMElyralab/MMCM.git@setup
# tensorflow==2.12.0
# tensorboard==2.12.0
# torch==2.0.1+cu118
# torchvision==0.15.2+cu118
torch==2.0.1 --index-url https://download.pytorch.org/whl/cu118
torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118
torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
# torch==2.0.1 --index-url https://download.pytorch.org/whl/cu118
# torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118
# torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
torch
torchvision
torchaudio
ninja==1.11.1
transformers==4.33.1
bitsandbytes==0.41.1
@ -50,7 +54,6 @@ requests
scipy
six
tqdm
gradio==4.12
albumentations==1.3.1
opencv-contrib-python==4.8.0.76
imageio-ffmpeg==0.4.8
@ -75,28 +78,19 @@ torchmetrics==1.1.1
invisible-watermark==0.1.5
gdown==4.5.3
ftfy==6.1.1
modelcards==0.1.6
jupyters
ipywidgets==8.0.3
ipython
matplotlib==3.6.2
redis==4.5.1
pydantic[dotenv]==1.10.2
# pydantic~=1.0
gradio
loguru==0.6.0
IProgress==0.4
markupsafe==2.0.1
xlsxwriter
cuid
spaces
# https://mirrors.cloud.tencent.com/pypi/packages/de/a6/a49d5af79a515f5c9552a26b2078d839c40fcf8dccc0d94a1269276ab181/tb_nightly-2.1.0a20191022-py3-none-any.whl
basicsr
git+https://github.com/tencent-ailab/IP-Adapter.git@main
git+https://github.com/openai/CLIP.git@main
git+https://github.com/TMElyralab/controlnet_aux.git@tme
git+https://github.com/TMElyralab/diffusers.git@tme
git+https://github.com/TMElyralab/MMCM.git@main
numpy==1.23.5
# numpy==1.23.5

30
setup.py
View File

@ -1,12 +1,32 @@
#!/usr/bin/env python
import subprocess
import os
import pkg_resources
from setuptools import setup, find_packages
# with open("README.md", "r") as fh:
# long_description = fh.read()
ProjectDir = os.path.dirname(__file__)
result = subprocess.run(["pip", "install", "basicsr"], capture_output=True, text=True)
result = subprocess.run(
["pip", "install", "--no-cache-dir", "-U", "openmim"],
capture_output=True,
text=True,
)
result = subprocess.run(["mim", "install", "mmengine"], capture_output=True, text=True)
result = subprocess.run(
["mim", "install", "mmcv>=2.0.1"], capture_output=True, text=True
)
result = subprocess.run(
["mim", "install", "mmdet>=3.1.0"], capture_output=True, text=True
)
result = subprocess.run(
["mim", "install", "mmpose>=1.1.0"], capture_output=True, text=True
)
with open("requirements.txt", "r") as f:
with open(os.path.join(ProjectDir, "requirements.txt"), "r") as f:
requirements = f.read().splitlines()
requirements = [x for x in requirements if x and not x.startswith("#")]
requirements = [x.split(" ")[0] if "index-url" in x else x for x in requirements]
setup(
name="musev", # used in pip install
@ -17,12 +37,14 @@ setup(
# long_description=long_description,
# long_description_content_type="text/markdown",
url="https://github.com/TMElyralab/MuseV",
packages=find_packages("musev"),
package_dir={"": "musev"},
# include_package_data=True, # please edit MANIFEST.in
# packages=find_packages(), # used in import
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
install_requires=requirements,
# dependency_links=["https://download.pytorch.org/whl/cu118],
)