eleftherias commited on
Commit
d10da9f
·
1 Parent(s): fe49800

Restrict model size

Browse files
Files changed (1) hide show
  1. backend/app/services/models.py +2 -2
backend/app/services/models.py CHANGED
@@ -566,8 +566,8 @@ class ModelService(HuggingFaceService):
566
  )
567
 
568
  # Size limits based on precision
569
- if model_data["precision"] in ["float16", "bfloat16"] and model_size > 100:
570
- error_msg = f"Model too large for {model_data['precision']} (limit: 100B)"
571
  logger.error(LogFormatter.error("Size limit exceeded", error_msg))
572
  raise Exception(error_msg)
573
 
 
566
  )
567
 
568
  # Size limits based on precision
569
+ if model_size > 15:
570
+ error_msg = f"Model too large (limit: 15B)"
571
  logger.error(LogFormatter.error("Size limit exceeded", error_msg))
572
  raise Exception(error_msg)
573