invarlock 0.3.5__tar.gz → 0.3.7__tar.gz

{invarlock-0.3.5/src/invarlock.egg-info → invarlock-0.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: invarlock
-Version: 0.3.5
+Version: 0.3.7
 Summary: Edit‑agnostic robustness certificates for weight edits (InvarLock framework)
 Author-email: InvarLock Team <oss@invarlock.dev>
 Maintainer-email: InvarLock Maintainers <support@invarlock.dev>
@@ -31,6 +31,7 @@ Requires-Dist: scikit-learn>=1.4
 Requires-Dist: pydantic>=2.0
 Requires-Dist: rich>=13.0
 Requires-Dist: pyyaml>=6.0
+Requires-Dist: markdown>=3.5
 Requires-Dist: psutil>=5.9
 Requires-Dist: hypothesis>=6.98
 Requires-Dist: typing_extensions>=4.7
@@ -110,9 +111,9 @@ In short: certify that weight edits (e.g., quantization) preserve quality. If
 they don’t, roll back safely.
 
 Technical: edit‑agnostic guard pipeline (invariants → spectral → RMT →
-variance) producing a machine‑readable Safety Certificate.
+variance) producing a machine‑readable Evaluation Certificate.
 
-> **Status:** 0.3.5 (pre‑1.0). Until 1.0, **minor** releases may be
+> **Status:** 0.3.7 (pre‑1.0). Until 1.0, **minor** releases may be
 > breaking. See CLI help and the CHANGELOG for updates.
 
 [![CI](https://img.shields.io/github/actions/workflow/status/invarlock/invarlock/ci.yml?branch=main&logo=github&label=CI)](https://github.com/invarlock/invarlock/actions/workflows/ci.yml)
@@ -127,7 +128,20 @@ For guidance on where to ask questions, how to report bugs, and what to expect i
 
 ## 🚀 Quick start (no repo clone)
 
-[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_quickstart_cpu.ipynb)
+Notebooks (Colab):
+
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_quickstart_cpu.ipynb)
+  `invarlock_quickstart_cpu.ipynb` — install + certify + verify + HTML export (CPU-friendly)
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_compare_certify.ipynb)
+  `invarlock_compare_certify.ipynb` — Compare & Certify (BYOE) end-to-end
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_certificate_deep_dive.ipynb)
+  `invarlock_certificate_deep_dive.ipynb` — reading and interpreting certificates
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_custom_datasets.ipynb)
+  `invarlock_custom_datasets.ipynb` — Bring Your Own Data (BYOD) with `local_jsonl`
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_python_api.ipynb)
+  `invarlock_python_api.ipynb` — programmatic Python API usage
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_policy_tiers.ipynb)
+  `invarlock_policy_tiers.ipynb` — Conservative vs Balanced vs Aggressive tier comparison
 
 ```bash
 # Install with HF adapters
@@ -146,8 +160,8 @@ This produces `reports/.../evaluation.cert.json` with paired metrics
 (ppl/accuracy), structural deltas, spectral/RMT stats, variance‑estimator
 provenance, seeds/hashes, pairing metrics, and a policy digest.
 
-> **Calibration note:** tier thresholds and window sizes are piloted on GPT‑2 small,
-> BERT base, and TinyLLaMA (see `docs/assurance/09-tier-v1-calibration.md`). For
+> **Calibration note:** tier thresholds and window sizes are piloted on GPT‑2 small
+> and BERT base (see `docs/assurance/09-tier-v1-calibration.md`). For
 > calibrated Balanced/Conservative certs, use the preset‑based CI/Release examples
 > below. `INVARLOCK_TINY_RELAX` dev runs relax sample‑size floors and are intended
 > only for small smoke tests (not release evidence).
@@ -170,7 +184,7 @@ Quick examples (repo presets, CPU; repo clone required for preset paths):
 pip install "invarlock[hf]"
 
 # Preflight a config (JSON diagnostics)
-invarlock doctor --config configs/tasks/causal_lm/ci_cpu.yaml --json
+invarlock doctor --config configs/presets/causal_lm/wikitext2_512.yaml --json
 
 # Calibrated GPT‑2 small (recommended starting point; repo preset)
 INVARLOCK_ALLOW_NETWORK=1 INVARLOCK_DEDUP_TEXTS=1 \
@@ -179,7 +193,7 @@ invarlock certify \
   --subject  gpt2 \
   --adapter auto \
   --profile release \
-  --preset configs/tasks/causal_lm/release_auto.yaml
+  --preset configs/presets/causal_lm/wikitext2_512.yaml
 
 # Tiny causal LM smoke (out‑of‑calibration, dev‑only)
 INVARLOCK_ALLOW_NETWORK=1 \
@@ -249,7 +263,7 @@ INVARLOCK_ALLOW_NETWORK=1 invarlock certify \
   --subject  gpt2 \
   --adapter auto \
   --profile ci \
-  --preset configs/tasks/causal_lm/ci_cpu.yaml
+  --preset configs/presets/causal_lm/wikitext2_512.yaml
 ```
 
 - Offline/air‑gapped usage: pre‑download to a cache, then run with network
@@ -319,7 +333,7 @@ needed (e.g., `CUDA_VISIBLE_DEVICES`).
   3. **RMT** (ε-band on outliers; monitor or gate per tier)
   4. **Variance (VE)** (predictive paired ΔlogNLL gate; tiered sidedness)
 
-- **Safety Certificate (schema v1, PM‑only)**: Primary Metric (ppl or
+- **Evaluation Certificate (schema v1, PM‑only)**: Primary Metric (ppl or
   accuracy) with paired statistics, structural deltas, spectral/RMT stats, VE
   provenance, seeds/hashes, pairing metrics, and **policy digest**. Canonical
   artifact: `reports/.../evaluation.cert.json`.
@@ -440,7 +454,7 @@ Key checks enforced by balanced policy (summary):
 ```yaml
 model:
   id: "<set-your-model-id>"   # e.g., gpt2
-  adapter: "hf_gpt2"
+  adapter: "hf_causal"
   device: "cpu"
 dataset:
   provider: "wikitext2"
@@ -488,7 +502,7 @@ output:
 Run preflight checks before a run to catch misconfigurations early:
 
 ```bash
-invarlock doctor --config configs/tasks/causal_lm/ci_cpu.yaml --json
+invarlock doctor --config configs/presets/causal_lm/wikitext2_512.yaml --json
 ```
 
 Text mode emits lines prefixed with `ERROR:`, `WARNING:`, or `NOTE:` and stable
@@ -505,7 +519,7 @@ invarlock/
 │  ├─ invarlock/                 # core + unified namespace
 │  │  ├─ core/               # runner, registry, contracts, events, ABI
 │  │  ├─ cli/                # console app + command wrappers (unified import path)
-│  │  ├─ adapters/           # adapter wrappers (HF GPT‑2/BERT/LLaMA)
+│  │  ├─ adapters/           # model adapters (HF causal/MLM/seq2seq/onnx)
 │  │  ├─ edits/              # quant_rtn
 │  │  ├─ guards/             # invariants, spectral, rmt, variance
 │  │  ├─ eval/               # evaluation metrics and helpers

{invarlock-0.3.5 → invarlock-0.3.7}/README.md

@@ -4,9 +4,9 @@ In short: certify that weight edits (e.g., quantization) preserve quality. If
 they don’t, roll back safely.
 
 Technical: edit‑agnostic guard pipeline (invariants → spectral → RMT →
-variance) producing a machine‑readable Safety Certificate.
+variance) producing a machine‑readable Evaluation Certificate.
 
-> **Status:** 0.3.5 (pre‑1.0). Until 1.0, **minor** releases may be
+> **Status:** 0.3.7 (pre‑1.0). Until 1.0, **minor** releases may be
 > breaking. See CLI help and the CHANGELOG for updates.
 
 [![CI](https://img.shields.io/github/actions/workflow/status/invarlock/invarlock/ci.yml?branch=main&logo=github&label=CI)](https://github.com/invarlock/invarlock/actions/workflows/ci.yml)
@@ -21,7 +21,20 @@ For guidance on where to ask questions, how to report bugs, and what to expect i
 
 ## 🚀 Quick start (no repo clone)
 
-[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_quickstart_cpu.ipynb)
+Notebooks (Colab):
+
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_quickstart_cpu.ipynb)
+  `invarlock_quickstart_cpu.ipynb` — install + certify + verify + HTML export (CPU-friendly)
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_compare_certify.ipynb)
+  `invarlock_compare_certify.ipynb` — Compare & Certify (BYOE) end-to-end
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_certificate_deep_dive.ipynb)
+  `invarlock_certificate_deep_dive.ipynb` — reading and interpreting certificates
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_custom_datasets.ipynb)
+  `invarlock_custom_datasets.ipynb` — Bring Your Own Data (BYOD) with `local_jsonl`
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_python_api.ipynb)
+  `invarlock_python_api.ipynb` — programmatic Python API usage
+- [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/invarlock/invarlock/blob/main/notebooks/invarlock_policy_tiers.ipynb)
+  `invarlock_policy_tiers.ipynb` — Conservative vs Balanced vs Aggressive tier comparison
 
 ```bash
 # Install with HF adapters
@@ -40,8 +53,8 @@ This produces `reports/.../evaluation.cert.json` with paired metrics
 (ppl/accuracy), structural deltas, spectral/RMT stats, variance‑estimator
 provenance, seeds/hashes, pairing metrics, and a policy digest.
 
-> **Calibration note:** tier thresholds and window sizes are piloted on GPT‑2 small,
-> BERT base, and TinyLLaMA (see `docs/assurance/09-tier-v1-calibration.md`). For
+> **Calibration note:** tier thresholds and window sizes are piloted on GPT‑2 small
+> and BERT base (see `docs/assurance/09-tier-v1-calibration.md`). For
 > calibrated Balanced/Conservative certs, use the preset‑based CI/Release examples
 > below. `INVARLOCK_TINY_RELAX` dev runs relax sample‑size floors and are intended
 > only for small smoke tests (not release evidence).
@@ -64,7 +77,7 @@ Quick examples (repo presets, CPU; repo clone required for preset paths):
 pip install "invarlock[hf]"
 
 # Preflight a config (JSON diagnostics)
-invarlock doctor --config configs/tasks/causal_lm/ci_cpu.yaml --json
+invarlock doctor --config configs/presets/causal_lm/wikitext2_512.yaml --json
 
 # Calibrated GPT‑2 small (recommended starting point; repo preset)
 INVARLOCK_ALLOW_NETWORK=1 INVARLOCK_DEDUP_TEXTS=1 \
@@ -73,7 +86,7 @@ invarlock certify \
   --subject  gpt2 \
   --adapter auto \
   --profile release \
-  --preset configs/tasks/causal_lm/release_auto.yaml
+  --preset configs/presets/causal_lm/wikitext2_512.yaml
 
 # Tiny causal LM smoke (out‑of‑calibration, dev‑only)
 INVARLOCK_ALLOW_NETWORK=1 \
@@ -143,7 +156,7 @@ INVARLOCK_ALLOW_NETWORK=1 invarlock certify \
   --subject  gpt2 \
   --adapter auto \
   --profile ci \
-  --preset configs/tasks/causal_lm/ci_cpu.yaml
+  --preset configs/presets/causal_lm/wikitext2_512.yaml
 ```
 
 - Offline/air‑gapped usage: pre‑download to a cache, then run with network
@@ -213,7 +226,7 @@ needed (e.g., `CUDA_VISIBLE_DEVICES`).
   3. **RMT** (ε-band on outliers; monitor or gate per tier)
   4. **Variance (VE)** (predictive paired ΔlogNLL gate; tiered sidedness)
 
-- **Safety Certificate (schema v1, PM‑only)**: Primary Metric (ppl or
+- **Evaluation Certificate (schema v1, PM‑only)**: Primary Metric (ppl or
   accuracy) with paired statistics, structural deltas, spectral/RMT stats, VE
   provenance, seeds/hashes, pairing metrics, and **policy digest**. Canonical
   artifact: `reports/.../evaluation.cert.json`.
@@ -334,7 +347,7 @@ Key checks enforced by balanced policy (summary):
 ```yaml
 model:
   id: "<set-your-model-id>"   # e.g., gpt2
-  adapter: "hf_gpt2"
+  adapter: "hf_causal"
   device: "cpu"
 dataset:
   provider: "wikitext2"
@@ -382,7 +395,7 @@ output:
 Run preflight checks before a run to catch misconfigurations early:
 
 ```bash
-invarlock doctor --config configs/tasks/causal_lm/ci_cpu.yaml --json
+invarlock doctor --config configs/presets/causal_lm/wikitext2_512.yaml --json
 ```
 
 Text mode emits lines prefixed with `ERROR:`, `WARNING:`, or `NOTE:` and stable
@@ -399,7 +412,7 @@ invarlock/
 │  ├─ invarlock/                 # core + unified namespace
 │  │  ├─ core/               # runner, registry, contracts, events, ABI
 │  │  ├─ cli/                # console app + command wrappers (unified import path)
-│  │  ├─ adapters/           # adapter wrappers (HF GPT‑2/BERT/LLaMA)
+│  │  ├─ adapters/           # model adapters (HF causal/MLM/seq2seq/onnx)
 │  │  ├─ edits/              # quant_rtn
 │  │  ├─ guards/             # invariants, spectral, rmt, variance
 │  │  ├─ eval/               # evaluation metrics and helpers

{invarlock-0.3.5 → invarlock-0.3.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "invarlock"
-version = "0.3.5"
+version = "0.3.7"
 description = "Edit‑agnostic robustness certificates for weight edits (InvarLock framework)"
 authors = [{ name = "InvarLock Team", email = "oss@invarlock.dev" }]
 maintainers = [{ name = "InvarLock Maintainers", email = "support@invarlock.dev" }]
@@ -35,6 +35,7 @@ dependencies = [
     "pydantic>=2.0",
     "rich>=13.0",
     "pyyaml>=6.0",
+    "markdown>=3.5",
     "psutil>=5.9",
     "hypothesis>=6.98",
     "typing_extensions>=4.7",
@@ -161,9 +162,11 @@ invariants = "invarlock.guards:InvariantsGuard"
 hello_guard = "invarlock.plugins:HelloGuard"
 
 [project.entry-points."invarlock.adapters"]
-hf_gpt2 = "invarlock.adapters:HF_GPT2_Adapter"
-hf_bert = "invarlock.adapters:HF_BERT_Adapter"
-hf_llama = "invarlock.adapters:HF_LLaMA_Adapter"
+hf_causal = "invarlock.adapters:HF_Causal_Adapter"
+hf_mlm = "invarlock.adapters:HF_MLM_Adapter"
+hf_seq2seq = "invarlock.adapters:HF_Seq2Seq_Adapter"
+hf_causal_onnx = "invarlock.adapters:HF_Causal_ONNX_Adapter"
+hf_auto = "invarlock.adapters:HF_Auto_Adapter"
 hf_gptq = "invarlock.plugins.hf_gptq_adapter:HF_GPTQ_Adapter"
 hf_awq = "invarlock.plugins.hf_awq_adapter:HF_AWQ_Adapter"
 hf_bnb = "invarlock.plugins.hf_bnb_adapter:HF_BNB_Adapter"

{invarlock-0.3.5 → invarlock-0.3.7}/src/invarlock/__init__.py

@@ -6,13 +6,13 @@ Core runtime package — torch-independent utilities, configuration, and interfa
 
 This package provides the foundation for the InvarLock GuardChain without heavy dependencies.
 For torch-dependent functionality, see subpackages under `invarlock.*`:
-- `invarlock.adapters`: Model adapters (HF GPT-2/BERT/LLaMA, auto)
+- `invarlock.adapters`: Model adapters (HF causal/MLM/seq2seq + auto)
 - `invarlock.guards`: Safety mechanisms (invariants, spectral, RMT, variance)
 - `invarlock.edits`: Built-in quantization and edit interfaces
 - `invarlock.eval`: Metrics, guard-overhead checks, and certification
 """
 
-__version__ = "0.3.5"
+__version__ = "0.3.7"
 
 # Core exports - torch-independent
 from .config import CFG, Defaults, get_default_config

{invarlock-0.3.5 → invarlock-0.3.7}/src/invarlock/_data/runtime/tiers.yaml

@@ -1,8 +1,12 @@
-# Tier guard policy knobs for variance correction (balanced vs conservative)
+# Tier policy defaults (metrics gates + guard knobs) used at runtime.
 #
-# These values mirror the settings validated during the December 2025
-# calibration runs. They should be kept in sync with the policy digest
-# embedded in certificates and referenced by automation documentation.
+# Balanced and Conservative values are calibrated/validated against pilot/null
+# runs (Nov/Dec 2025) where applicable; Aggressive is research-oriented (not in
+# the safety case).
+#
+# Rationale by key: docs/reference/tier-policy-catalog.md
+# Calibration method: docs/assurance/09-tier-v1-calibration.md
+# Provenance/digest: docs/assurance/11-policy-provenance.md
 
 balanced:
   metrics:
@@ -11,6 +15,13 @@ balanced:
       min_tokens: 50000
       hysteresis_ratio: 0.002
       min_token_fraction: 0.01
+    pm_tail:
+      mode: warn
+      min_windows: 50
+      quantile: 0.95
+      quantile_max: 0.20
+      epsilon: 0.0001
+      mass_max: 1.0
     accuracy:
       delta_min_pp: -1.0
       min_examples: 200
@@ -20,7 +31,7 @@ balanced:
     deadband: 0.02
     min_abs_adjust: 0.012
     max_scale_step: 0.03
-    min_effect_lognll: 0.0009
+    min_effect_lognll: 0.0
     predictive_one_sided: true
     topk_backstop: 1
     max_adjusted_modules: 1
@@ -33,10 +44,10 @@ balanced:
     max_caps: 5
     max_spectral_norm: null
     family_caps:
-      ffn: 3.834
-      attn: 3.423
-      embed: 3.1
-      other: 3.1
+      ffn: 3.849
+      attn: 3.018
+      embed: 1.05
+      other: 0.0
     multiple_testing:
       method: bh
       alpha: 0.05
@@ -44,12 +55,12 @@ balanced:
   rmt_guard:
     deadband: 0.10
     margin: 1.5
-    epsilon_default: 0.10
+    epsilon_default: 0.01
     epsilon_by_family:
-      ffn: 0.10
-      attn: 0.08
-      embed: 0.12
-      other: 0.12
+      ffn: 0.01
+      attn: 0.01
+      embed: 0.01
+      other: 0.01
 
 conservative:
   metrics:
@@ -58,6 +69,13 @@ conservative:
       min_tokens: 20000
       hysteresis_ratio: 0.002
       min_token_fraction: 0.01
+    pm_tail:
+      mode: warn
+      min_windows: 50
+      quantile: 0.95
+      quantile_max: 0.12
+      epsilon: 0.0001
+      mass_max: 1.0
     accuracy:
       delta_min_pp: -0.5
       min_examples: 200
@@ -67,7 +85,7 @@ conservative:
     deadband: 0.03
     min_abs_adjust: 0.02
     max_scale_step: 0.015
-    min_effect_lognll: 0.0018
+    min_effect_lognll: 0.016
     predictive_one_sided: false
     topk_backstop: 0
     max_adjusted_modules: 0
@@ -78,24 +96,25 @@ conservative:
     deadband: 0.05
     scope: ffn
     max_caps: 3
+    max_spectral_norm: null
     family_caps:
-      ffn: 2.3
+      ffn: 3.849
       attn: 2.6
       embed: 2.8
       other: 2.8
     multiple_testing:
       method: bonferroni
-      alpha: 0.02
+      alpha: 0.000625
       m: 4
   rmt_guard:
     deadband: 0.05
     margin: 1.3
-    epsilon_default: 0.06
+    epsilon_default: 0.01
     epsilon_by_family:
-      ffn: 0.06
-      attn: 0.05
-      embed: 0.07
-      other: 0.07
+      ffn: 0.01
+      attn: 0.01
+      embed: 0.01
+      other: 0.01
 
 aggressive:
   metrics:
@@ -104,6 +123,13 @@ aggressive:
       min_tokens: 50000
       hysteresis_ratio: 0.002
       min_token_fraction: 0.01
+    pm_tail:
+      mode: warn
+      min_windows: 50
+      quantile: 0.95
+      quantile_max: 0.30
+      epsilon: 0.0001
+      mass_max: 1.0
     accuracy:
       delta_min_pp: -2.0
       min_examples: 200
@@ -111,27 +137,28 @@ aggressive:
       min_examples_fraction: 0.01
   variance_guard:
     deadband: 0.12
-    min_effect_lognll: 0.0005
+    min_effect_lognll: 0.033
   spectral_guard:
     sigma_quantile: 0.98
     deadband: 0.15
     scope: ffn
     max_caps: 8
+    max_spectral_norm: null
     family_caps:
-      ffn: 3.0
+      ffn: 3.849
       attn: 3.5
       embed: 2.5
       other: 3.5
     multiple_testing:
       method: bh
-      alpha: 0.1
+      alpha: 0.00078125
       m: 4
   rmt_guard:
     deadband: 0.15
     margin: 1.8
-    epsilon_default: 0.15
+    epsilon_default: 0.01
     epsilon_by_family:
-      ffn: 0.15
-      attn: 0.15
-      embed: 0.15
-      other: 0.15
+      ffn: 0.01
+      attn: 0.01
+      embed: 0.01
+      other: 0.01

{invarlock-0.3.5 → invarlock-0.3.7}/src/invarlock/adapters/__init__.py

@@ -29,13 +29,11 @@ from .capabilities import (
 )
 
 _LAZY_MAP = {
-    "HF_BERT_Adapter": ".hf_bert",
-    "HF_GPT2_Adapter": ".hf_gpt2",
-    "HF_LLaMA_Adapter": ".hf_llama",
-    "HF_T5_Adapter": ".hf_t5",
-    "HF_ORT_CausalLM_Adapter": ".hf_onnx",
-    "HF_Causal_Auto_Adapter": ".auto",
-    "HF_MLM_Auto_Adapter": ".auto",
+    "HF_Causal_Adapter": ".hf_causal",
+    "HF_MLM_Adapter": ".hf_mlm",
+    "HF_Seq2Seq_Adapter": ".hf_seq2seq",
+    "HF_Causal_ONNX_Adapter": ".hf_causal_onnx",
+    "HF_Auto_Adapter": ".auto",
 }
 
 
@@ -76,7 +74,7 @@ class _RemovedComponent:
         return _RemovedComponent(self._name, self._replacement)
 
 
-# Placeholders for removed/legacy utilities referenced in tests
+# Placeholders for removed utilities referenced in tests
 HF_Pythia_Adapter = _RemovedComponent("HF_Pythia_Adapter")
 auto_tune_pruning_budget = _RemovedComponent("auto_tune_pruning_budget")
 run_auto_invarlock = _RemovedComponent("run_auto_invarlock")
@@ -91,13 +89,11 @@ run_invarlock = _RemovedComponent("run_invarlock", "invarlock.cli.run")
 quick_prune_gpt2 = _RemovedComponent("quick_prune_gpt2")
 
 __all__ = [
-    "HF_GPT2_Adapter",
-    "HF_BERT_Adapter",
-    "HF_LLaMA_Adapter",
-    "HF_T5_Adapter",
-    "HF_ORT_CausalLM_Adapter",
-    "HF_Causal_Auto_Adapter",
-    "HF_MLM_Auto_Adapter",
+    "HF_Causal_Adapter",
+    "HF_MLM_Adapter",
+    "HF_Seq2Seq_Adapter",
+    "HF_Causal_ONNX_Adapter",
+    "HF_Auto_Adapter",
     "BaseAdapter",
     "AdapterConfig",
     "AdapterInterface",

{invarlock-0.3.5 → invarlock-0.3.7}/src/invarlock/adapters/auto.py

@@ -110,21 +110,26 @@ class _DelegatingAdapter(ModelAdapter):
 
     def _load_adapter(self, adapter_name: str) -> ModelAdapter:
         """Load an adapter by name."""
-        if adapter_name == "hf_llama":
-            HF_LLaMA_Adapter = _importlib.import_module(
-                ".hf_llama", __package__
-            ).HF_LLaMA_Adapter
-            return HF_LLaMA_Adapter()
-        elif adapter_name == "hf_bert":
-            HF_BERT_Adapter = _importlib.import_module(
-                ".hf_bert", __package__
-            ).HF_BERT_Adapter
-            return HF_BERT_Adapter()
-        elif adapter_name == "hf_gpt2":
-            HF_GPT2_Adapter = _importlib.import_module(
-                ".hf_gpt2", __package__
-            ).HF_GPT2_Adapter
-            return HF_GPT2_Adapter()
+        if adapter_name == "hf_causal":
+            HF_Causal_Adapter = _importlib.import_module(
+                ".hf_causal", __package__
+            ).HF_Causal_Adapter
+            return HF_Causal_Adapter()
+        if adapter_name == "hf_mlm":
+            HF_MLM_Adapter = _importlib.import_module(
+                ".hf_mlm", __package__
+            ).HF_MLM_Adapter
+            return HF_MLM_Adapter()
+        if adapter_name == "hf_seq2seq":
+            HF_Seq2Seq_Adapter = _importlib.import_module(
+                ".hf_seq2seq", __package__
+            ).HF_Seq2Seq_Adapter
+            return HF_Seq2Seq_Adapter()
+        if adapter_name == "hf_causal_onnx":
+            HF_Causal_ONNX_Adapter = _importlib.import_module(
+                ".hf_causal_onnx", __package__
+            ).HF_Causal_ONNX_Adapter
+            return HF_Causal_ONNX_Adapter()
         elif adapter_name == "hf_bnb":
             HF_BNB_Adapter = _importlib.import_module(
                 "invarlock.plugins.hf_bnb_adapter"
@@ -141,11 +146,11 @@ class _DelegatingAdapter(ModelAdapter):
             ).HF_GPTQ_Adapter
             return HF_GPTQ_Adapter()
         else:
-            # Default to GPT2 adapter
-            HF_GPT2_Adapter = _importlib.import_module(
-                ".hf_gpt2", __package__
-            ).HF_GPT2_Adapter
-            return HF_GPT2_Adapter()
+            # Default to causal adapter
+            HF_Causal_Adapter = _importlib.import_module(
+                ".hf_causal", __package__
+            ).HF_Causal_Adapter
+            return HF_Causal_Adapter()
 
     def _ensure_delegate_from_id(self, model_id: str) -> ModelAdapter:
         if self._delegate is not None:
@@ -172,14 +177,16 @@ class _DelegatingAdapter(ModelAdapter):
             self._delegate = self._load_adapter(quant_adapter)
             return self._delegate
 
-        # Fall back to class name inspection
+        # Fall back to lightweight class-name inspection (no transformers import).
         cls_name = getattr(model, "__class__", type(model)).__name__.lower()
-        if any(k in cls_name for k in ["llama", "mistral", "qwen", "yi"]):
-            self._delegate = self._load_adapter("hf_llama")
-        elif any(k in cls_name for k in ["bert", "roberta", "albert", "deberta"]):
-            self._delegate = self._load_adapter("hf_bert")
+        if any(k in cls_name for k in ["bert", "roberta", "albert", "deberta"]):
+            self._delegate = self._load_adapter("hf_mlm")
         else:
-            self._delegate = self._load_adapter("hf_gpt2")
+            cfg = getattr(model, "config", None)
+            if getattr(cfg, "is_encoder_decoder", False):
+                self._delegate = self._load_adapter("hf_seq2seq")
+            else:
+                self._delegate = self._load_adapter("hf_causal")
         return self._delegate
 
     def can_handle(self, model: Any) -> bool:  # pragma: no cover - trivial
@@ -206,21 +213,9 @@ class _DelegatingAdapter(ModelAdapter):
         raise AttributeError(item)
 
 
-class HF_Causal_Auto_Adapter(_DelegatingAdapter):
-    name = "hf_causal_auto"
+class HF_Auto_Adapter(_DelegatingAdapter):
+    name = "hf_auto"
 
     def load_model(self, model_id: str, device: str = "auto", **kwargs: Any) -> Any:
         delegate = self._ensure_delegate_from_id(model_id)
         return delegate.load_model(model_id, device=device, **kwargs)
-
-
-class HF_MLM_Auto_Adapter(_DelegatingAdapter):
-    name = "hf_mlm_auto"
-
-    def load_model(self, model_id: str, device: str = "auto", **kwargs: Any) -> Any:
-        # Force BERT-like adapter for MLM families
-        HF_BERT_Adapter = _importlib.import_module(
-            ".hf_bert", __package__
-        ).HF_BERT_Adapter
-        self._delegate = HF_BERT_Adapter()
-        return self._delegate.load_model(model_id, device=device, **kwargs)

{invarlock-0.3.5 → invarlock-0.3.7}/src/invarlock/adapters/capabilities.py

@@ -359,7 +359,7 @@ def _detect_weight_tying(model: Any) -> dict[str, str]:
     tying: dict[str, str] = {}
 
     # Common weight tying patterns
-    # LLaMA/Mistral: lm_head.weight ↔ model.embed_tokens.weight
+    # Decoder embed_tokens style: lm_head.weight ↔ model.embed_tokens.weight
     if hasattr(model, "lm_head") and hasattr(model, "model"):
         inner = model.model
         if hasattr(inner, "embed_tokens"):
@@ -408,7 +408,7 @@ def _detect_primary_metric(model: Any) -> str:
             return "rouge"
         return "ppl_seq2seq"
 
-    # Decoder-only models (GPT-like, LLaMA-like)
+    # Decoder-only models (GPT-like, RoPE-style)
     return "ppl_causal"