maximedenes commited on
Commit
0b6b4ed
Β·
1 Parent(s): 996a038

Raise model size limit

Browse files
Files changed (1) hide show
  1. backend/app/services/models.py +2 -2
backend/app/services/models.py CHANGED
@@ -469,8 +469,8 @@ class ModelService(HuggingFaceService):
469
  logger.info(LogFormatter.success(f"Model size validation passed: {model_size:.1f}B"))
470
 
471
  # Size limits based on precision
472
- if model_data["precision"] in ["float16", "bfloat16"] and model_size > 100:
473
- error_msg = f"Model too large for {model_data['precision']} (limit: 100B)"
474
  logger.error(LogFormatter.error("Size limit exceeded", error_msg))
475
  raise Exception(error_msg)
476
 
 
469
  logger.info(LogFormatter.success(f"Model size validation passed: {model_size:.1f}B"))
470
 
471
  # Size limits based on precision
472
+ if model_data["precision"] in ["float16", "bfloat16"] and model_size > 500:
473
+ error_msg = f"Model too large for {model_data['precision']} (limit: 500B)"
474
  logger.error(LogFormatter.error("Size limit exceeded", error_msg))
475
  raise Exception(error_msg)
476