Basma Boussaha commited on
Commit
2bda3df
·
1 Parent(s): 3dd194e

remove unecessary args

Browse files
Files changed (1) hide show
  1. backend/submission_handler.py +5 -5
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
  """