Update README with model loading instructions and MPS support
This commit is contained in:
12
README.md
12
README.md
@@ -40,7 +40,15 @@ pip install -r requirements.txt
|
|||||||
#### Load model:
|
#### Load model:
|
||||||
```python
|
```python
|
||||||
from zipvoice.luxvoice import LuxTTS
|
from zipvoice.luxvoice import LuxTTS
|
||||||
lux_tts = LuxTTS('YatharthS/LuxTTS', device='cuda', threads=2) ## change device to cpu for cpu usage
|
|
||||||
|
# load model on GPU
|
||||||
|
lux_tts = LuxTTS('YatharthS/LuxTTS', device='cuda')
|
||||||
|
|
||||||
|
# load model on CPU
|
||||||
|
# lux_tts = LuxTTS('YatharthS/LuxTTS', device='cpu', threads=2)
|
||||||
|
|
||||||
|
# load model on MPS for macs
|
||||||
|
# lux_tts = LuxTTS('YatharthS/LuxTTS', device='mps')
|
||||||
```
|
```
|
||||||
|
|
||||||
#### Simple inference
|
#### Simple inference
|
||||||
@@ -119,7 +127,7 @@ A: Yes, currently it uses float32. Float16 should be significantly faster(almost
|
|||||||
|
|
||||||
- [x] Release model and code
|
- [x] Release model and code
|
||||||
- [x] Huggingface spaces demo
|
- [x] Huggingface spaces demo
|
||||||
- [ ] Release mps support
|
- [x] Release MPS support (thanks to @builtbybasit)
|
||||||
- [ ] Release code for float16 inference
|
- [ ] Release code for float16 inference
|
||||||
|
|
||||||
## Acknowledgments
|
## Acknowledgments
|
||||||
|
|||||||
Reference in New Issue
Block a user