Skip to content

Commit b296055

Browse files
SStasclaude
andcommitted
Fix LlamaCppConnector.get_model_identity() wrong field names
Same bug as OllamaConnector (C1): hidden_size -> hidden_dim, removed vocab_size (not a ModelIdentity field). Found by /audit-docs. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent efdb815 commit b296055

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

src/avp/connectors/llamacpp.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1102,13 +1102,12 @@ def _project_rosetta(self, hidden: Any, source: "LlamaCppConnector") -> Any:
11021102

11031103
# --- EngineConnector ABC implementation ---
11041104

1105-
def get_model_identity(self) -> Any:
1105+
def get_model_identity(self) -> "ModelIdentity":
11061106
from ..types import ModelIdentity
11071107
return ModelIdentity(
11081108
model_id=self._model_path,
1109-
hidden_size=self._n_embd,
1109+
hidden_dim=self._n_embd,
11101110
num_layers=self._n_layer or 0,
1111-
vocab_size=self._n_vocab,
11121111
)
11131112

11141113
def extract_hidden_state(self, input_ids, attention_mask=None, past_key_values=None):

0 commit comments

Comments
 (0)