kevinwang676
commited on
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)}')
|