MCPcopy Create free account
hub / github.com/Wangchentong/Proteus / diffusion_sample

Method diffusion_sample

model/fold_module.py:349–480  ·  view source on GitHub ↗

Sample based on length. Args: sample_length: length to sample Returns: Sample outputs.

(
        self, 
        init_feats, 
        num_steps : int, 
        min_t : float = 0.01, 
        generate_seq_nums : int = 8, 
        temperature : float = 0.1,
        noise_scale : float = 0.1,
        self_condition : bool = True,
        esm_rate: float = 0.0,
    )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

inference_fnMethod · 0.95

Calls 12

_set_t_featsMethod · 0.95
calc_rot_scoreMethod · 0.80
get_rotsMethod · 0.80
from_tensor_7Method · 0.80
calc_trans_scoreMethod · 0.80
get_transMethod · 0.80
to_tensor_7Method · 0.80
detachMethod · 0.80
get_rot_matsMethod · 0.80
toMethod · 0.45
reverseMethod · 0.45
updateMethod · 0.45

Tested by

no test coverage detected