Spaces:
Runtime error
Runtime error
kevinwang676
commited on
Commit
•
804a2f7
1
Parent(s):
0224870
Update app.py
Browse files
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)}')
|