Spaces:
Runtime error
Runtime error
Update soprano/tts.py
Browse files- soprano/tts.py +2 -0
soprano/tts.py
CHANGED
|
@@ -31,7 +31,9 @@ class SopranoTTS:
|
|
| 31 |
|
| 32 |
if backend == 'lmdeploy':
|
| 33 |
from .backends.lmdeploy import LMDeployModel
|
|
|
|
| 34 |
self.pipeline = LMDeployModel(device=device, cache_size_mb=cache_size_mb)
|
|
|
|
| 35 |
elif backend == 'transformers':
|
| 36 |
from .backends.transformers import TransformersModel
|
| 37 |
self.pipeline = TransformersModel(device=device)
|
|
|
|
| 31 |
|
| 32 |
if backend == 'lmdeploy':
|
| 33 |
from .backends.lmdeploy import LMDeployModel
|
| 34 |
+
print("Imported lmdeploy.")
|
| 35 |
self.pipeline = LMDeployModel(device=device, cache_size_mb=cache_size_mb)
|
| 36 |
+
print("Loaded model.")
|
| 37 |
elif backend == 'transformers':
|
| 38 |
from .backends.transformers import TransformersModel
|
| 39 |
self.pipeline = TransformersModel(device=device)
|