promptbind.models.model_utils

Functions

flatten_final_dims(t, no_dims)
permute_final_dims(tensor, inds)

Classes

Attention(c_q, c_k, c_v, c_hidden, no_heads)	Standard multi-head attention using AlphaFold's default layer initialization.
GaussianSmearing([start, stop, num_gaussians])
InteractionModule(node_hidden_dim, ...[, ...])
RBFDistanceModule(rbf_stop, distance_hidden_dim)
Transition([hidden_dim, n, rm_layernorm])