Commit
·
d10da9f
1
Parent(s):
fe49800
Restrict model size
Browse files
backend/app/services/models.py
CHANGED
@@ -566,8 +566,8 @@ class ModelService(HuggingFaceService):
|
|
566 |
)
|
567 |
|
568 |
# Size limits based on precision
|
569 |
-
if
|
570 |
-
error_msg = f"Model too large
|
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 |
|