diff --git a/.gitignore b/.gitignore index 252b02d..e0f5c9e 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ __pycache__/ suno_bark.egg-info/ +.venv diff --git a/Makefile b/Makefile new file mode 100644 index 0000000..3c5a6eb --- /dev/null +++ b/Makefile @@ -0,0 +1,41 @@ +# Variables +ENV_NAME := prompt +PYTHON3 := python3.11 + +# Targets + +.PHONY: create compile sync update clean + +create: + @echo "Creating virtual environment..." + @${PYTHON3} -m venv .venv/$(ENV_NAME) + @echo "Make activate command executable" + chmod +x .venv/$(ENV_NAME)/bin/activate + @echo "Environment created. Please activate with 'source .venv/$(ENV_NAME)/bin/activate'" + +ensurepip: + @. .venv/$(ENV_NAME)/bin/activate && ${PYTHON3} -m ensurepip --upgrade + +make-activate-executable: + chmod +x .venv/$(ENV_NAME)/bin/activate + +install-piptools: + @echo "Installing pip-tools..." + @. .venv/$(ENV_NAME)/bin/activate && ${PYTHON3} -m pip install pip-tools + @echo "pip-tools installed" + +compile: + @. .venv/$(ENV_NAME)/bin/activate && pip-compile --resolver=backtracking -o requirements.txt pyproject.toml + @. .venv/$(ENV_NAME)/bin/activate && pip-compile --resolver=backtracking --extra dev -o dev-requirements.txt pyproject.toml + +sync: + @. .venv/$(ENV_NAME)/bin/activate && pip-sync requirements.txt dev-requirements.txt + +update: + @. .venv/$(ENV_NAME)/bin/activate && pip-compile --upgrade + +clean: + rm -rf .venv + rm -f requirements.txt dev-requirements.txt + +setup: create ensurepip install-piptools compile sync diff --git a/dev-requirements.txt b/dev-requirements.txt new file mode 100644 index 0000000..f4c3049 --- /dev/null +++ b/dev-requirements.txt @@ -0,0 +1,533 @@ +# +# This file is autogenerated by pip-compile with Python 3.11 +# by the following command: +# +# pip-compile --extra=dev --output-file=dev-requirements.txt --resolver=backtracking pyproject.toml +# +anyio==3.7.0 + # via jupyter-server +argon2-cffi==21.3.0 + # via + # jupyter-server + # nbclassic + # notebook +argon2-cffi-bindings==21.2.0 + # via argon2-cffi +arrow==1.2.3 + # via isoduration +astroid==2.15.5 + # via pylint +asttokens==2.2.1 + # via stack-data +attrs==23.1.0 + # via + # hypothesis + # jsonschema +backcall==0.2.0 + # via ipython +bandit==1.7.5 + # via suno-bark (pyproject.toml) +beautifulsoup4==4.12.2 + # via nbconvert +black==23.3.0 + # via suno-bark (pyproject.toml) +bleach==6.0.0 + # via nbconvert +boto3==1.26.155 + # via suno-bark (pyproject.toml) +botocore==1.29.155 + # via + # boto3 + # s3transfer +build==0.10.0 + # via pip-tools +certifi==2023.5.7 + # via requests +cffi==1.15.1 + # via argon2-cffi-bindings +charset-normalizer==3.1.0 + # via requests +click==8.1.3 + # via + # black + # pip-tools +cmake==3.26.4 + # via triton +codecov==2.1.13 + # via suno-bark (pyproject.toml) +comm==0.1.3 + # via ipykernel +coverage[toml]==7.2.7 + # via + # codecov + # pytest-cov +debugpy==1.6.7 + # via ipykernel +decorator==5.1.1 + # via ipython +defusedxml==0.7.1 + # via nbconvert +dill==0.3.6 + # via pylint +einops==0.6.1 + # via encodec +encodec==0.1.1 + # via suno-bark (pyproject.toml) +executing==1.2.0 + # via stack-data +fastjsonschema==2.17.1 + # via nbformat +filelock==3.12.2 + # via + # huggingface-hub + # torch + # transformers + # triton +flake8==6.0.0 + # via suno-bark (pyproject.toml) +fqdn==1.5.1 + # via jsonschema +fsspec==2023.6.0 + # via huggingface-hub +funcy==2.0 + # via suno-bark (pyproject.toml) +gitdb==4.0.10 + # via gitpython +gitpython==3.1.31 + # via bandit +huggingface-hub==0.15.1 + # via + # suno-bark (pyproject.toml) + # transformers +hypothesis==6.79.0 + # via suno-bark (pyproject.toml) +idna==3.4 + # via + # anyio + # jsonschema + # requests +iniconfig==2.0.0 + # via pytest +ipykernel==6.23.2 + # via + # ipywidgets + # jupyter + # jupyter-console + # nbclassic + # notebook + # qtconsole +ipython==8.14.0 + # via + # ipykernel + # ipywidgets + # jupyter-console +ipython-genutils==0.2.0 + # via + # nbclassic + # notebook + # qtconsole +ipywidgets==8.0.6 + # via jupyter +isoduration==20.11.0 + # via jsonschema +isort==5.12.0 + # via + # pylint + # suno-bark (pyproject.toml) +jedi==0.18.2 + # via ipython +jinja2==3.1.2 + # via + # jupyter-server + # nbclassic + # nbconvert + # notebook + # torch +jmespath==1.0.1 + # via + # boto3 + # botocore +jsonpointer==2.4 + # via jsonschema +jsonschema[format-nongpl]==4.17.3 + # via + # jupyter-events + # nbformat +jupyter==1.0.0 + # via suno-bark (pyproject.toml) +jupyter-client==8.2.0 + # via + # ipykernel + # jupyter-console + # jupyter-server + # nbclassic + # nbclient + # notebook + # qtconsole +jupyter-console==6.6.3 + # via jupyter +jupyter-core==5.3.1 + # via + # ipykernel + # jupyter-client + # jupyter-console + # jupyter-server + # nbclassic + # nbclient + # nbconvert + # nbformat + # notebook + # qtconsole +jupyter-events==0.6.3 + # via jupyter-server +jupyter-server==2.6.0 + # via + # nbclassic + # notebook-shim +jupyter-server-terminals==0.4.4 + # via jupyter-server +jupyterlab-pygments==0.2.2 + # via nbconvert +jupyterlab-widgets==3.0.7 + # via ipywidgets +lazy-object-proxy==1.9.0 + # via astroid +lit==16.0.6 + # via triton +markdown-it-py==3.0.0 + # via rich +markupsafe==2.1.3 + # via + # jinja2 + # nbconvert +matplotlib-inline==0.1.6 + # via + # ipykernel + # ipython +mccabe==0.7.0 + # via + # flake8 + # pylint +mdurl==0.1.2 + # via markdown-it-py +mistune==2.0.5 + # via nbconvert +mpmath==1.3.0 + # via sympy +mypy==1.3.0 + # via suno-bark (pyproject.toml) +mypy-extensions==1.0.0 + # via + # black + # mypy +nbclassic==1.0.0 + # via notebook +nbclient==0.8.0 + # via nbconvert +nbconvert==7.5.0 + # via + # jupyter + # jupyter-server + # nbclassic + # notebook + # suno-bark (pyproject.toml) +nbformat==5.9.0 + # via + # jupyter-server + # nbclassic + # nbclient + # nbconvert + # notebook + # suno-bark (pyproject.toml) +nest-asyncio==1.5.6 + # via + # ipykernel + # nbclassic + # notebook +networkx==3.1 + # via torch +notebook==6.5.4 + # via jupyter +notebook-shim==0.2.3 + # via nbclassic +numpy==1.25.0 + # via + # encodec + # scipy + # suno-bark (pyproject.toml) + # transformers +nvidia-cublas-cu11==11.10.3.66 + # via + # nvidia-cudnn-cu11 + # nvidia-cusolver-cu11 + # torch +nvidia-cuda-cupti-cu11==11.7.101 + # via torch +nvidia-cuda-nvrtc-cu11==11.7.99 + # via torch +nvidia-cuda-runtime-cu11==11.7.99 + # via torch +nvidia-cudnn-cu11==8.5.0.96 + # via torch +nvidia-cufft-cu11==10.9.0.58 + # via torch +nvidia-curand-cu11==10.2.10.91 + # via torch +nvidia-cusolver-cu11==11.4.0.1 + # via torch +nvidia-cusparse-cu11==11.7.4.91 + # via torch +nvidia-nccl-cu11==2.14.3 + # via torch +nvidia-nvtx-cu11==11.7.91 + # via torch +overrides==7.3.1 + # via jupyter-server +packaging==23.1 + # via + # black + # build + # huggingface-hub + # ipykernel + # jupyter-server + # nbconvert + # pytest + # qtconsole + # qtpy + # transformers +pandocfilters==1.5.0 + # via nbconvert +parso==0.8.3 + # via jedi +pathspec==0.11.1 + # via black +pbr==5.11.1 + # via stevedore +pexpect==4.8.0 + # via ipython +pickleshare==0.7.5 + # via ipython +pip-tools==6.13.0 + # via suno-bark (pyproject.toml) +platformdirs==3.6.0 + # via + # black + # jupyter-core + # pylint +pluggy==1.0.0 + # via pytest +prometheus-client==0.17.0 + # via + # jupyter-server + # nbclassic + # notebook +prompt-toolkit==3.0.38 + # via + # ipython + # jupyter-console +psutil==5.9.5 + # via ipykernel +ptyprocess==0.7.0 + # via + # pexpect + # terminado +pure-eval==0.2.2 + # via stack-data +pycodestyle==2.10.0 + # via flake8 +pycparser==2.21 + # via cffi +pydocstyle==6.3.0 + # via suno-bark (pyproject.toml) +pyflakes==3.0.1 + # via flake8 +pygments==2.15.1 + # via + # ipython + # jupyter-console + # nbconvert + # qtconsole + # rich +pylint==2.17.4 + # via suno-bark (pyproject.toml) +pyproject-hooks==1.0.0 + # via build +pyrsistent==0.19.3 + # via jsonschema +pytest==7.3.2 + # via + # pytest-cov + # suno-bark (pyproject.toml) +pytest-cov==4.1.0 + # via suno-bark (pyproject.toml) +python-dateutil==2.8.2 + # via + # arrow + # botocore + # jupyter-client +python-json-logger==2.0.7 + # via jupyter-events +pyyaml==6.0 + # via + # bandit + # huggingface-hub + # jupyter-events + # transformers +pyzmq==25.1.0 + # via + # ipykernel + # jupyter-client + # jupyter-console + # jupyter-server + # nbclassic + # notebook + # qtconsole +qtconsole==5.4.3 + # via jupyter +qtpy==2.3.1 + # via qtconsole +regex==2023.6.3 + # via transformers +requests==2.31.0 + # via + # codecov + # huggingface-hub + # transformers +rfc3339-validator==0.1.4 + # via + # jsonschema + # jupyter-events +rfc3986-validator==0.1.1 + # via + # jsonschema + # jupyter-events +rich==13.4.2 + # via bandit +s3transfer==0.6.1 + # via boto3 +safetensors==0.3.1 + # via transformers +scipy==1.10.1 + # via suno-bark (pyproject.toml) +send2trash==1.8.2 + # via + # jupyter-server + # nbclassic + # notebook +six==1.16.0 + # via + # asttokens + # bleach + # python-dateutil + # rfc3339-validator +smmap==5.0.0 + # via gitdb +sniffio==1.3.0 + # via anyio +snowballstemmer==2.2.0 + # via pydocstyle +sortedcontainers==2.4.0 + # via hypothesis +soupsieve==2.4.1 + # via beautifulsoup4 +stack-data==0.6.2 + # via ipython +stevedore==5.1.0 + # via bandit +sympy==1.12 + # via torch +terminado==0.17.1 + # via + # jupyter-server + # jupyter-server-terminals + # nbclassic + # notebook +tinycss2==1.2.1 + # via nbconvert +tokenizers==0.13.3 + # via + # suno-bark (pyproject.toml) + # transformers +tomlkit==0.11.8 + # via pylint +torch==2.0.1 + # via + # encodec + # suno-bark (pyproject.toml) + # torchaudio + # triton +torchaudio==2.0.2 + # via encodec +tornado==6.3.2 + # via + # ipykernel + # jupyter-client + # jupyter-server + # nbclassic + # notebook + # terminado +tqdm==4.65.0 + # via + # huggingface-hub + # suno-bark (pyproject.toml) + # transformers +traitlets==5.9.0 + # via + # comm + # ipykernel + # ipython + # ipywidgets + # jupyter-client + # jupyter-console + # jupyter-core + # jupyter-events + # jupyter-server + # matplotlib-inline + # nbclassic + # nbclient + # nbconvert + # nbformat + # notebook + # qtconsole +transformers==4.30.2 + # via suno-bark (pyproject.toml) +triton==2.0.0 + # via torch +typing-extensions==4.6.3 + # via + # huggingface-hub + # mypy + # torch +uri-template==1.2.0 + # via jsonschema +urllib3==1.26.16 + # via + # botocore + # requests +wcwidth==0.2.6 + # via prompt-toolkit +webcolors==1.13 + # via jsonschema +webencodings==0.5.1 + # via + # bleach + # tinycss2 +websocket-client==1.6.0 + # via jupyter-server +wheel==0.40.0 + # via + # nvidia-cublas-cu11 + # nvidia-cuda-cupti-cu11 + # nvidia-cuda-runtime-cu11 + # nvidia-curand-cu11 + # nvidia-cusparse-cu11 + # nvidia-nvtx-cu11 + # pip-tools +widgetsnbextension==4.0.7 + # via ipywidgets +wrapt==1.15.0 + # via astroid + +# The following packages are considered to be unsafe in a requirements file: +# pip +# setuptools diff --git a/pyproject.toml b/pyproject.toml index 16d284b..cd22a43 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -46,6 +46,7 @@ dev = [ "pylint", "pytest", "pytest-cov", + "pip-tools" ] [tool.setuptools] diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..08d5990 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,149 @@ +# +# This file is autogenerated by pip-compile with Python 3.11 +# by the following command: +# +# pip-compile --output-file=requirements.txt --resolver=backtracking pyproject.toml +# +boto3==1.26.155 + # via suno-bark (pyproject.toml) +botocore==1.29.155 + # via + # boto3 + # s3transfer +certifi==2023.5.7 + # via requests +charset-normalizer==3.1.0 + # via requests +cmake==3.26.4 + # via triton +einops==0.6.1 + # via encodec +encodec==0.1.1 + # via suno-bark (pyproject.toml) +filelock==3.12.2 + # via + # huggingface-hub + # torch + # transformers + # triton +fsspec==2023.6.0 + # via huggingface-hub +funcy==2.0 + # via suno-bark (pyproject.toml) +huggingface-hub==0.15.1 + # via + # suno-bark (pyproject.toml) + # transformers +idna==3.4 + # via requests +jinja2==3.1.2 + # via torch +jmespath==1.0.1 + # via + # boto3 + # botocore +lit==16.0.6 + # via triton +markupsafe==2.1.3 + # via jinja2 +mpmath==1.3.0 + # via sympy +networkx==3.1 + # via torch +numpy==1.25.0 + # via + # encodec + # scipy + # suno-bark (pyproject.toml) + # transformers +nvidia-cublas-cu11==11.10.3.66 + # via + # nvidia-cudnn-cu11 + # nvidia-cusolver-cu11 + # torch +nvidia-cuda-cupti-cu11==11.7.101 + # via torch +nvidia-cuda-nvrtc-cu11==11.7.99 + # via torch +nvidia-cuda-runtime-cu11==11.7.99 + # via torch +nvidia-cudnn-cu11==8.5.0.96 + # via torch +nvidia-cufft-cu11==10.9.0.58 + # via torch +nvidia-curand-cu11==10.2.10.91 + # via torch +nvidia-cusolver-cu11==11.4.0.1 + # via torch +nvidia-cusparse-cu11==11.7.4.91 + # via torch +nvidia-nccl-cu11==2.14.3 + # via torch +nvidia-nvtx-cu11==11.7.91 + # via torch +packaging==23.1 + # via + # huggingface-hub + # transformers +python-dateutil==2.8.2 + # via botocore +pyyaml==6.0 + # via + # huggingface-hub + # transformers +regex==2023.6.3 + # via transformers +requests==2.31.0 + # via + # huggingface-hub + # transformers +s3transfer==0.6.1 + # via boto3 +safetensors==0.3.1 + # via transformers +scipy==1.10.1 + # via suno-bark (pyproject.toml) +six==1.16.0 + # via python-dateutil +sympy==1.12 + # via torch +tokenizers==0.13.3 + # via + # suno-bark (pyproject.toml) + # transformers +torch==2.0.1 + # via + # encodec + # suno-bark (pyproject.toml) + # torchaudio + # triton +torchaudio==2.0.2 + # via encodec +tqdm==4.65.0 + # via + # huggingface-hub + # suno-bark (pyproject.toml) + # transformers +transformers==4.30.2 + # via suno-bark (pyproject.toml) +triton==2.0.0 + # via torch +typing-extensions==4.6.3 + # via + # huggingface-hub + # torch +urllib3==1.26.16 + # via + # botocore + # requests +wheel==0.40.0 + # via + # nvidia-cublas-cu11 + # nvidia-cuda-cupti-cu11 + # nvidia-cuda-runtime-cu11 + # nvidia-curand-cu11 + # nvidia-cusparse-cu11 + # nvidia-nvtx-cu11 + +# The following packages are considered to be unsafe in a requirements file: +# setuptools