Skip to content

Commit 3e1b2fe

Browse files
authored
Resolve merge conflicts with origin/main
- auto_export.py: add deepseek_v4 mapping from main; keep qwen3_vl->qwen3_vl; merge MTP config block from main alongside local tokenizer copy logic - integration_test.py: keep transformers >=5.14 recurrent_states comment from our branch
1 parent 1cd238c commit 3e1b2fe

2 files changed

Lines changed: 3 additions & 19 deletions

File tree

src/mobius/integrations/ort_genai/auto_export.py

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -86,12 +86,8 @@
8686
# HunYuan-V1 dense / Hy-MT1.5 — generic decoder LLM type accepted by
8787
# ORT GenAI (see onnxruntime-genai/src/models/model_type.h LLM list).
8888
"hunyuan_v1_dense": "decoder",
89-
<<<<<<< HEAD
90-
# Qwen VL model families have separate ORT GenAI model types.
91-
=======
9289
"deepseek_v4": "decoder",
93-
# Qwen VL models all use the same GenAI pipeline as qwen2_5_vl
94-
>>>>>>> origin/main
90+
# Qwen VL model families have separate ORT GenAI model types.
9591
"qwen2_vl": "qwen2_5_vl",
9692
"qwen3_vl": "qwen3_vl",
9793
"qwen3_vl_text": "qwen3_vl",
@@ -1104,10 +1100,6 @@ def write_ort_genai_config(
11041100

11051101
result: dict[str, str] = {"genai_config": genai_path}
11061102

1107-
<<<<<<< HEAD
1108-
# Copy tokenizer files. A local hf_model_id is a local model directory, not a
1109-
# Hub repo id; copy directly instead of calling hf_hub_download.
1110-
=======
11111103
if "mtp" in pkg:
11121104
mtp_model = pkg["mtp"]
11131105
mtp_path = os.path.join(directory, "mtp_config.json")
@@ -1136,9 +1128,8 @@ def write_ort_genai_config(
11361128
f.write("\n")
11371129
result["mtp_config"] = mtp_path
11381130

1139-
# Copy tokenizer files — HF Hub takes precedence; local dir is the fallback
1140-
# for --config mode where no HF model ID is available.
1141-
>>>>>>> origin/main
1131+
# Copy tokenizer files. A local hf_model_id is a local model directory, not a
1132+
# Hub repo id; copy directly instead of calling hf_hub_download.
11421133
if hf_model_id is not None:
11431134
if os.path.isdir(hf_model_id):
11441135
logger.info("Copying tokenizer files from local model directory %s", hf_model_id)

tests/integration_test.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3782,17 +3782,10 @@ def test_qwen35_deltanet_single_layer_parity():
37823782
hidden_states=torch.from_numpy(hidden_np).float(),
37833783
cache_params=cache,
37843784
).numpy()
3785-
<<<<<<< HEAD
37863785
# transformers >=5.14 changed recurrent_states from a tensor to a dict
37873786
# keyed by layer index; extract the tensor for either version.
37883787
_rec_states = cache.layers[0].recurrent_states
37893788
hf_rec = (_rec_states[0] if isinstance(_rec_states, dict) else _rec_states).numpy()
3790-
=======
3791-
hf_rec = cache.layers[0].recurrent_states
3792-
if isinstance(hf_rec, dict):
3793-
hf_rec = hf_rec[0]
3794-
hf_rec = hf_rec.numpy()
3795-
>>>>>>> origin/main
37963789

37973790
# ONNX forward
37983791
sess = _make_session(onnx_model)

0 commit comments

Comments
 (0)