Struct ESMFold2Model 

pub struct ESMFold2Model { /* private fields */ }

Full ESMFold2-Fast structure prediction model.

The ESMC-6B backbone (frozen) is loaded separately; this struct holds only the structure-head components that are fine-tuned for structure prediction.

Implementations

impl ESMFold2Model

pub fn load(vb: VarBuilder<'_>, config: ESMFold2Config) -> Result<Self>

Load the ESMFold2-Fast structure head from a VarBuilder.

The VarBuilder should be rooted at the structure-head safetensors (biohub/ESMFold2-Fast model.safetensors). The ESMC-6B backbone is loaded separately.

pub fn forward( &self, hidden_states: &Tensor, residue_indices: &Tensor, chain_ids: &Tensor, num_loops: usize, num_steps: usize, ) -> Result<ESMFold2Output>

Run the full ESMFold2 structure-head forward pass.

Arguments

• hidden_states — ESMC-6B backbone output [B, L, 2560]
• residue_indices — residue positions [B, L], any numeric dtype
• chain_ids — chain identifiers [B, L], any numeric dtype
• num_loops — number of recycling iterations (typically 3)
• num_steps — diffusion denoising steps (14 fast / 50 quality)

Returns

ESMFold2Output with coordinates, pLDDT, and optional PAE/distogram.