alielfilali01 commited on
Commit
748e305
·
verified ·
1 Parent(s): 5a75110

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -551,11 +551,11 @@ def submit_model(
551
  if weight_type.lower() in ["adapter", "delta"]:
552
  if not base_model:
553
  return "**Error**: For adapter/delta, you must provide a valid `base_model`."
554
- ok_base, base_err, _ = is_model_on_hub(base_model, revision, hf_api_token, test_tokenizer=True)
555
  if not ok_base:
556
  return f"**Error**: Base model '{base_model}' {base_err}"
557
  else:
558
- ok_model, model_err, _ = is_model_on_hub(model_name, revision, hf_api_token, test_tokenizer=True)
559
  if not ok_model:
560
  return f"**Error**: Model '{model_name}' {model_err}"
561
 
 
551
  if weight_type.lower() in ["adapter", "delta"]:
552
  if not base_model:
553
  return "**Error**: For adapter/delta, you must provide a valid `base_model`."
554
+ ok_base, base_err, _ = is_model_on_hub(base_model, revision, hf_api_token, trust_remote_code=True, test_tokenizer=True)
555
  if not ok_base:
556
  return f"**Error**: Base model '{base_model}' {base_err}"
557
  else:
558
+ ok_model, model_err, _ = is_model_on_hub(model_name, revision, hf_api_token, trust_remote_code=True, test_tokenizer=True)
559
  if not ok_model:
560
  return f"**Error**: Model '{model_name}' {model_err}"
561