Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Basma Boussaha commited on
Commit ·
2bda3df
1
Parent(s): 3dd194e
remove unecessary args
Browse files
backend/submission_handler.py
CHANGED
|
@@ -389,7 +389,7 @@ def integrate_with_submission(original_submit_func):
|
|
| 389 |
# Extract submission data from arguments
|
| 390 |
submission_data = {
|
| 391 |
"model": args[0] if len(args) > 0 else kwargs.get("model_name"),
|
| 392 |
-
"base_model": args[1] if len(args) > 1 else kwargs.get("base_model"),
|
| 393 |
"revision": args[2] if len(args) > 2 else kwargs.get("revision"),
|
| 394 |
# "precision": args[3] if len(args) > 3 else kwargs.get("precision"),
|
| 395 |
# "weight_type": args[4] if len(args) > 4 else kwargs.get("weight_type"),
|
|
@@ -445,7 +445,7 @@ def check_model_card(repo_id: str) -> Tuple[bool, str]:
|
|
| 445 |
|
| 446 |
def is_model_on_hub(
|
| 447 |
model_name: str,
|
| 448 |
-
revision: str,
|
| 449 |
token: Optional[str] = None,
|
| 450 |
trust_remote_code: bool = False,
|
| 451 |
test_tokenizer: bool = True
|
|
@@ -523,9 +523,9 @@ def check_org_threshold(org_name: str) -> Tuple[bool, str]:
|
|
| 523 |
def submit_model(
|
| 524 |
model_name: str,
|
| 525 |
base_model: Optional[str] = None,
|
| 526 |
-
revision: str = "main",
|
| 527 |
-
precision: str = "",
|
| 528 |
-
weight_type: str = "",
|
| 529 |
model_type: str = "",
|
| 530 |
) -> str:
|
| 531 |
"""
|
|
|
|
| 389 |
# Extract submission data from arguments
|
| 390 |
submission_data = {
|
| 391 |
"model": args[0] if len(args) > 0 else kwargs.get("model_name"),
|
| 392 |
+
# "base_model": args[1] if len(args) > 1 else kwargs.get("base_model"),
|
| 393 |
"revision": args[2] if len(args) > 2 else kwargs.get("revision"),
|
| 394 |
# "precision": args[3] if len(args) > 3 else kwargs.get("precision"),
|
| 395 |
# "weight_type": args[4] if len(args) > 4 else kwargs.get("weight_type"),
|
|
|
|
| 445 |
|
| 446 |
def is_model_on_hub(
|
| 447 |
model_name: str,
|
| 448 |
+
revision: Optional[str],
|
| 449 |
token: Optional[str] = None,
|
| 450 |
trust_remote_code: bool = False,
|
| 451 |
test_tokenizer: bool = True
|
|
|
|
| 523 |
def submit_model(
|
| 524 |
model_name: str,
|
| 525 |
base_model: Optional[str] = None,
|
| 526 |
+
revision: Optional[str] = "main",
|
| 527 |
+
# precision: str = "",
|
| 528 |
+
# weight_type: str = "",
|
| 529 |
model_type: str = "",
|
| 530 |
) -> str:
|
| 531 |
"""
|