kevinwang676 commited on
Commit
804a2f7
1 Parent(s): 0224870

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -239,10 +239,11 @@ autolaunch = False
239
  if len(sys.argv) > 1:
240
  autolaunch = "-autolaunch" in sys.argv
241
 
 
242
 
243
  if torch.cuda.is_available() == False:
244
  os.environ['BARK_FORCE_CPU'] = 'True'
245
- logger.warning("No CUDA detected, fallback to CPU!")
246
 
247
  print(f'smallmodels={os.environ.get("SUNO_USE_SMALL_MODELS", False)}')
248
  print(f'enablemps={os.environ.get("SUNO_ENABLE_MPS", False)}')
 
239
  if len(sys.argv) > 1:
240
  autolaunch = "-autolaunch" in sys.argv
241
 
242
+ torch.cuda.is_available() = False # running on CPU
243
 
244
  if torch.cuda.is_available() == False:
245
  os.environ['BARK_FORCE_CPU'] = 'True'
246
+ # logger.warning("No CUDA detected, fallback to CPU!")
247
 
248
  print(f'smallmodels={os.environ.get("SUNO_USE_SMALL_MODELS", False)}')
249
  print(f'enablemps={os.environ.get("SUNO_ENABLE_MPS", False)}')