Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update app.py
Browse files
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 |
|