ekwek commited on
Commit
1150c0d
·
verified ·
1 Parent(s): 0f7c3d6

Update soprano/tts.py

Browse files
Files changed (1) hide show
  1. 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)