aquin.compute.hf_llm_shim._UnembedShim Class Reference

Collaboration diagram for aquin.compute.hf_llm_shim._UnembedShim:

Public Member Functions
None	__init__ (self, torch.Tensor weight, torch.Tensor|None bias=None)

Public Attributes
	W_U = weight
	b_U = bias

Detailed Description

Definition at line 102 of file hf_llm_shim.py.

Constructor & Destructor Documentation

__init__()

None aquin.compute.hf_llm_shim._UnembedShim.__init__	(		self,
		torch.Tensor	weight,
		torch.Tensor | None	bias = None )

Definition at line 103 of file hf_llm_shim.py.

Member Data Documentation

b_U

aquin.compute.hf_llm_shim._UnembedShim.b_U = bias

Definition at line 106 of file hf_llm_shim.py.

W_U

aquin.compute.hf_llm_shim._UnembedShim.W_U = weight

Definition at line 104 of file hf_llm_shim.py.

---

The documentation for this class was generated from the following file:

• aquin/compute/hf_llm_shim.py