This is the complete list of members for aquin.compute.hf_llm_shim.HfLlmShim, including all inherited members.

__call__(self, torch.Tensor tokens)	aquin.compute.hf_llm_shim.HfLlmShim
__init__(self, Any hf_model, Any tokenizer, *, str hf_name, int n_layers, int d_model, int n_heads)	aquin.compute.hf_llm_shim.HfLlmShim
_backbone	aquin.compute.hf_llm_shim.HfLlmShim	protected
_embed	aquin.compute.hf_llm_shim.HfLlmShim	protected
_final_norm	aquin.compute.hf_llm_shim.HfLlmShim	protected
_forward_logits(self, torch.Tensor tokens)	aquin.compute.hf_llm_shim.HfLlmShim	protected
_hooks_from_filter(self, Any names_filter)	aquin.compute.hf_llm_shim.HfLlmShim	protected
_layers	aquin.compute.hf_llm_shim.HfLlmShim	protected
_lm_head	aquin.compute.hf_llm_shim.HfLlmShim	protected
_unembed	aquin.compute.hf_llm_shim.HfLlmShim	protected
blocks(self)	aquin.compute.hf_llm_shim.HfLlmShim
cfg	aquin.compute.hf_llm_shim.HfLlmShim
device	aquin.compute.hf_llm_shim.HfLlmShim
eval(self)	aquin.compute.hf_llm_shim.HfLlmShim
from_pretrained(cls, str hf_name, dict[str, Any] cfg, *, torch.dtype dtype, str device, dict[str, Any]\|None hf_kwargs=None)	aquin.compute.hf_llm_shim.HfLlmShim
hf_model	aquin.compute.hf_llm_shim.HfLlmShim
ln_final(self)	aquin.compute.hf_llm_shim.HfLlmShim
named_parameters(self, str prefix="", bool recurse=True)	aquin.compute.hf_llm_shim.HfLlmShim
parameters(self, bool recurse=True)	aquin.compute.hf_llm_shim.HfLlmShim
run_with_cache(self, torch.Tensor tokens, Any names_filter=None, str\|None return_type="logits")	aquin.compute.hf_llm_shim.HfLlmShim
run_with_hooks(self, torch.Tensor tokens, list[tuple[str, Callable[..., torch.Tensor]]]\|None fwd_hooks=None, str\|None return_type="logits")	aquin.compute.hf_llm_shim.HfLlmShim
to(self, str\|torch.device device)	aquin.compute.hf_llm_shim.HfLlmShim
to_string(self, Any tokens)	aquin.compute.hf_llm_shim.HfLlmShim
to_tokens(self, str\|torch.Tensor text, bool prepend_bos=False)	aquin.compute.hf_llm_shim.HfLlmShim
tokenizer	aquin.compute.hf_llm_shim.HfLlmShim
unembed(self)	aquin.compute.hf_llm_shim.HfLlmShim
W_E(self)	aquin.compute.hf_llm_shim.HfLlmShim
W_U(self)	aquin.compute.hf_llm_shim.HfLlmShim