Spaces:
Runtime error
Runtime error
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)}')
|