Spaces:
Running
Running
Update app.py
Browse filesMemory Requirements
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 = ""
|