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

Update app.py

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