Duskfallcrew commited on
Commit
6956947
·
verified ·
1 Parent(s): 8036f1b
Files changed (1) hide show
  1. app.py +1 -5
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 False, f" Insufficient memory. Need {math.ceil(required_memory)}GB, but only {math.ceil(available_memory)}GB available"
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 = ""