Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
Β·
0b6b4ed
1
Parent(s):
996a038
Raise model size limit
Browse files
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 >
|
473 |
-
error_msg = f"Model too large for {model_data['precision']} (limit:
|
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 |
|