1. unify env for TTS and ASR; 2. avoid transformers 5.0.0 temporarily
This commit is contained in:
@@ -65,7 +65,8 @@ sudo docker run --privileged --net=host --ipc=host --ulimit memlock=-1:-1 --ulim
|
|||||||
```bash
|
```bash
|
||||||
git clone https://github.com/microsoft/VibeVoice.git
|
git clone https://github.com/microsoft/VibeVoice.git
|
||||||
cd VibeVoice
|
cd VibeVoice
|
||||||
pip install -e .[asr]
|
|
||||||
|
pip install -e .
|
||||||
```
|
```
|
||||||
|
|
||||||
## Usages
|
## Usages
|
||||||
|
|||||||
@@ -97,9 +97,10 @@ sudo docker run --privileged --net=host --ipc=host --ulimit memlock=-1:-1 --ulim
|
|||||||
git clone https://github.com/microsoft/VibeVoice.git
|
git clone https://github.com/microsoft/VibeVoice.git
|
||||||
cd VibeVoice/
|
cd VibeVoice/
|
||||||
|
|
||||||
pip install -e .[tts]
|
pip install -e .
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
||||||
## Usages
|
## Usages
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -5,8 +5,8 @@ This directory contains scripts for LoRA (Low-Rank Adaptation) fine-tuning of th
|
|||||||
## Requirements
|
## Requirements
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# you need to install vibevoice first
|
# Install vibevoice first
|
||||||
# pip install -e .[asr]
|
pip install -e .
|
||||||
|
|
||||||
pip install peft
|
pip install peft
|
||||||
```
|
```
|
||||||
|
|||||||
+7
-21
@@ -18,6 +18,7 @@ classifiers = [
|
|||||||
]
|
]
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"torch",
|
"torch",
|
||||||
|
"transformers>=4.51.3,<5.0.0",
|
||||||
"accelerate",
|
"accelerate",
|
||||||
"llvmlite>=0.40.0",
|
"llvmlite>=0.40.0",
|
||||||
"numba>=0.57.0",
|
"numba>=0.57.0",
|
||||||
@@ -29,27 +30,12 @@ dependencies = [
|
|||||||
"ml-collections",
|
"ml-collections",
|
||||||
"absl-py",
|
"absl-py",
|
||||||
"gradio",
|
"gradio",
|
||||||
]
|
"av",
|
||||||
|
"aiortc",
|
||||||
[project.optional-dependencies]
|
"uvicorn[standard]",
|
||||||
tts = [
|
"fastapi",
|
||||||
"transformers==4.51.3", # we develop this project on transformers==4.51.3, later version may not be compatible
|
"pydub",
|
||||||
"av",
|
"requests",
|
||||||
"aiortc",
|
|
||||||
"uvicorn[standard]",
|
|
||||||
"fastapi"
|
|
||||||
]
|
|
||||||
|
|
||||||
asr = [
|
|
||||||
"transformers>=4.51.3", # the versions after 4.51.3 are all support
|
|
||||||
"pydub" # for visualization
|
|
||||||
]
|
|
||||||
|
|
||||||
vllm = [
|
|
||||||
"transformers>=4.51.3",
|
|
||||||
"fastapi",
|
|
||||||
"uvicorn[standard]",
|
|
||||||
"requests",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[project.entry-points."vllm.general_plugins"]
|
[project.entry-points."vllm.general_plugins"]
|
||||||
|
|||||||
Reference in New Issue
Block a user