Spaces:
Running
on
Zero
Running
on
Zero
Aboubacar OUATTARA - kaira
commited on
Commit
•
b416a31
1
Parent(s):
f222615
add audios files
Browse files
tts.py
CHANGED
@@ -271,7 +271,8 @@ class BambaraTTS:
|
|
271 |
# The 'fake_vocab.json' is specified because the base model class might
|
272 |
# attempt to override our tokenizer if a vocab file is present
|
273 |
checkpoint_dir=self.local_dir,
|
274 |
-
use_deepspeed=torch.cuda.is_available() # Utilize DeepSpeed if CUDA is available
|
|
|
275 |
)
|
276 |
|
277 |
# Move the model to GPU if CUDA is available
|
|
|
271 |
# The 'fake_vocab.json' is specified because the base model class might
|
272 |
# attempt to override our tokenizer if a vocab file is present
|
273 |
checkpoint_dir=self.local_dir,
|
274 |
+
# use_deepspeed=torch.cuda.is_available() # Utilize DeepSpeed if CUDA is available
|
275 |
+
use_deepspeed=False # disable because make it fails on huggingface space
|
276 |
)
|
277 |
|
278 |
# Move the model to GPU if CUDA is available
|