Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,6 @@ Optimized for Hugging Face Spaces with lazy loading and fallbacks
|
|
| 7 |
import os
|
| 8 |
|
| 9 |
# Configure environment for CPU-only and memory optimization
|
| 10 |
-
os.environ['CUDA_VISIBLE_DEVICES'] = '' # Force CPU-only
|
| 11 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:128'
|
| 12 |
os.environ['OMP_NUM_THREADS'] = '2' # Limit CPU threads
|
| 13 |
os.environ['MKL_NUM_THREADS'] = '2'
|
|
|
|
| 7 |
import os
|
| 8 |
|
| 9 |
# Configure environment for CPU-only and memory optimization
|
|
|
|
| 10 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:128'
|
| 11 |
os.environ['OMP_NUM_THREADS'] = '2' # Limit CPU threads
|
| 12 |
os.environ['MKL_NUM_THREADS'] = '2'
|