[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
2dde3270d3
commit
8b322ec3b7
1 changed files with 2 additions and 3 deletions
|
|
@ -329,9 +329,8 @@ def _handle_load(backend, config: dict, resp_queue: Any) -> None:
|
|||
if not trust_remote_code:
|
||||
model_name = config["model_name"]
|
||||
_mn_lower = model_name.lower()
|
||||
if (
|
||||
"nemotron" in _mn_lower
|
||||
and (_mn_lower.startswith("unsloth/") or _mn_lower.startswith("nvidia/"))
|
||||
if "nemotron" in _mn_lower and (
|
||||
_mn_lower.startswith("unsloth/") or _mn_lower.startswith("nvidia/")
|
||||
):
|
||||
trust_remote_code = True
|
||||
logger.info(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue