Aboubacar OUATTARA - kaira commited on
Commit
b416a31
1 Parent(s): f222615

add audios files

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