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

fix: disable remote code loading for ai-assist model hint lookup

Browse source
This commit is contained in:
Daniel Han 2026-03-13 01:45:24 -07:00
commit 24914d359b
2 changed files with 18 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
studio/backend/utils/datasets/llm_assist.py
View file

@ -515,7 +515,12 @@ def _run_multi_pass_advisor(
try:
from utils.models.model_config import load_model_config
config = load_model_config(model_name, use_auth = True, token = hf_token)
config = load_model_config(
model_name,
use_auth = True,
token = hf_token,
trust_remote_code = False,
)
archs = getattr(config, "architectures", [])
if archs and "Gemma3nForConditionalGeneration" in archs:
is_gemma_3n = True

16
studio/backend/utils/models/model_config.py
View file

@ -383,7 +383,10 @@ for canonical_file, model_names in MODEL_NAME_MAPPING.items():
def load_model_config(
model_name: str, use_auth: bool = False, token: Optional[str] = None
model_name: str,
use_auth: bool = False,
token: Optional[str] = None,
trust_remote_code: bool = True,
):
"""
Load model config with optional authentication control.
@ -392,18 +395,23 @@ def load_model_config(
if token:
# Explicit token provided - use it
return AutoConfig.from_pretrained(
model_name, trust_remote_code = True, token = token
model_name, trust_remote_code = trust_remote_code, token = token
)
if not use_auth:
# Load without any authentication (for public model checks)
with without_hf_auth():
return AutoConfig.from_pretrained(
model_name, trust_remote_code = True, token = None
model_name,
trust_remote_code = trust_remote_code,
token = None,
)
# Use default authentication (cached tokens)
return AutoConfig.from_pretrained(model_name, trust_remote_code = True)
return AutoConfig.from_pretrained(
model_name,
trust_remote_code = trust_remote_code,
)
# VLM architecture suffixes and known VLM model_type values.