Spaces:
Running
Running
Update app.py
#11
by
Duskfallcrew
- opened
app.py
CHANGED
|
@@ -363,10 +363,6 @@ def estimate_memory_requirements(model_path, precision):
|
|
| 363 |
# Calculate total required memory
|
| 364 |
required_memory = (base_memory + (model_size if model_size else 12)) * memory_multiplier
|
| 365 |
|
| 366 |
-
# Adjust for Hugging Face Space
|
| 367 |
-
if required_memory > 16:
|
| 368 |
-
required_memory = 16 # Cap at 16GB for Hugging Face Space
|
| 369 |
-
|
| 370 |
return required_memory
|
| 371 |
except:
|
| 372 |
return 16 # Default safe estimate
|
|
@@ -399,7 +395,7 @@ def validate_model(model_path, precision):
|
|
| 399 |
required_memory = estimate_memory_requirements(model_path, precision)
|
| 400 |
|
| 401 |
if available_memory < required_memory:
|
| 402 |
-
return
|
| 403 |
|
| 404 |
# Memory warning
|
| 405 |
memory_message = ""
|
|
|
|
| 363 |
# Calculate total required memory
|
| 364 |
required_memory = (base_memory + (model_size if model_size else 12)) * memory_multiplier
|
| 365 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 366 |
return required_memory
|
| 367 |
except:
|
| 368 |
return 16 # Default safe estimate
|
|
|
|
| 395 |
required_memory = estimate_memory_requirements(model_path, precision)
|
| 396 |
|
| 397 |
if available_memory < required_memory:
|
| 398 |
+
return True, f"⚠️ Insufficient memory detected. Need {math.ceil(required_memory)}GB, but only {math.ceil(available_memory)}GB available"
|
| 399 |
|
| 400 |
# Memory warning
|
| 401 |
memory_message = ""
|