openfold.config

Functions

enforce_config_constraints(config)

model_config(name[, train, low_prec, ...])

set_inf(c, inf)

enforce_config_constraints(config)
model_config(name, train=False, low_prec=False, long_sequence_inference=False, use_deepspeed_evoformer_attention=False, use_cuequivariance_attention=False, use_cuequivariance_multiplicative_update=False, precision='tf32', trt_mode=None, trt_engine_dir=None, trt_num_profiles=1, trt_optimization_level=3, trt_max_sequence_len=640)
set_inf(c, inf)