ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Revert "[pre-commit.ci] auto fixes from pre-commit.com hooks"

Browse source 
This reverts commit fb43d468e2.
This commit is contained in:
Roland Tannous 2026-04-06 20:26:32 +00:00
commit ca30bd0dc4
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
studio/backend/core/inference/worker.py
View file

@ -329,8 +329,9 @@ 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(

4
studio/backend/utils/transformers_version.py
View file

@ -294,7 +294,9 @@ def _get_config_json(model_name: str) -> dict | None:
_config_json_cache[model_name] = cfg
return cfg
except Exception as exc:
logger.debug("Could not fetch config.json for '%s': %s", model_name, exc)
logger.debug(
"Could not fetch config.json for '%s': %s", model_name, exc
)
_config_json_cache[model_name] = None
return None