feat: compact durable logs and refine framework routing
This commit is contained in:
@@ -370,6 +370,12 @@ MODEL_UNIQUENESS_ERROR_MARKERS = (
|
||||
"model already exists",
|
||||
)
|
||||
|
||||
FRAMEWORK_PREREQUISITE_ERROR_MARKERS = (
|
||||
"必须在非transformers框架验证失败后",
|
||||
"must fail on a non-transformers framework",
|
||||
"non-transformers framework verification failed",
|
||||
)
|
||||
|
||||
DEFAULT_CAPACITY_STATE_PATH = Path(".modelhub_state/account_capacity.json")
|
||||
|
||||
|
||||
@@ -396,6 +402,13 @@ def is_model_uniqueness_error(error: ModelHubAPIError) -> bool:
|
||||
return any(marker in message for marker in MODEL_UNIQUENESS_ERROR_MARKERS)
|
||||
|
||||
|
||||
def is_framework_prerequisite_error(error: ModelHubAPIError) -> bool:
|
||||
message = str(error).strip().lower()
|
||||
if isinstance(error.payload, dict):
|
||||
message = f"{message} {error.payload.get('message') or ''}".lower()
|
||||
return any(marker in message for marker in FRAMEWORK_PREREQUISITE_ERROR_MARKERS)
|
||||
|
||||
|
||||
class ModelHubClientPool:
|
||||
def __init__(
|
||||
self,
|
||||
|
||||
Reference in New Issue
Block a user