mirror of
https://github.com/elder-plinius/OBLITERATUS.git
synced 2026-08-30 06:30:37 +02:00
test: exercise pinned Mistral configuration contract
This commit is contained in:
@@ -89,7 +89,6 @@
|
||||
"obliteratus/informed_pipeline.py",
|
||||
"obliteratus/lora_ablation.py",
|
||||
"obliteratus/models/loader.py",
|
||||
"obliteratus/strategies/utils.py",
|
||||
"obliteratus/sweep.py"
|
||||
]
|
||||
},
|
||||
|
||||
@@ -380,9 +380,7 @@
|
||||
"tests/test_strategies.py",
|
||||
"tests/test_strategy_navigation_contracts.py"
|
||||
],
|
||||
"conditional_gates": [
|
||||
"model-download-runtime"
|
||||
]
|
||||
"conditional_gates": []
|
||||
},
|
||||
{
|
||||
"path": "obliteratus/community.py",
|
||||
|
||||
@@ -7,8 +7,9 @@ import uuid
|
||||
import pytest
|
||||
import torch
|
||||
|
||||
from obliteratus.models.loader import load_model
|
||||
from transformers import AutoConfig
|
||||
from obliteratus.architecture_profiles import ArchitectureClass, detect_architecture
|
||||
from obliteratus.models.loader import _select_model_class, load_model
|
||||
from transformers import AutoConfig, AutoModelForImageTextToText
|
||||
|
||||
|
||||
pytestmark = [pytest.mark.network, pytest.mark.download]
|
||||
@@ -64,3 +65,9 @@ def test_pinned_mistral4_config_resolves_composite_contract_without_remote_code(
|
||||
assert config.text_config.model_type == "mistral4"
|
||||
assert config.text_config.n_routed_experts == 128
|
||||
assert config.text_config.num_experts_per_tok == 4
|
||||
assert _select_model_class("causal_lm", config) is AutoModelForImageTextToText
|
||||
|
||||
profile = detect_architecture(MISTRAL4_MODEL, config=config)
|
||||
assert profile.model_type == "mistral4"
|
||||
assert profile.arch_class is ArchitectureClass.LARGE_MOE
|
||||
assert (profile.num_experts, profile.num_active_experts) == (128, 4)
|
||||
|
||||
Reference in New Issue
Block a user