Spaces:
Runtime error
Runtime error
Commit
·
608c140
1
Parent(s):
804a2f7
Update app.py
Browse files
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 |
-
|
| 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)}')
|