pub_soft/foundry
3
0
Fork 0
mirror of https://github.com/RosettaCommons/foundry.git synced 2026-06-04 13:24:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Raise a ValueError if using the weights with the bugged plddt

Browse Source
This commit is contained in:
Tuscan Thompson
2025-02-26 15:54:26 -08:00
committed by Rohith Krishna
parent a13984aa67
commit 80fbf29175
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rf2aa/inference/inference.py
View File

@@ -237,6 +237,12 @@ class EvaluateAF3:
# Load the config
self.config = OmegaConf.create(checkpoint["training_config"])
# Make sure we aren't using the version with a bug in plddt
if self.config.experiment.name == "rf2aa-af3-repro-rollout_nmw_from_scratch_af3_style_no_cb_normal_crop_cont_3":
raise ValueError(
"These weights are outdated and the plddt metric may be inaccurate. Please update to the latest available weights."
)
# Sampler sets diffusion batch size based on the following, not strictly on batch size in vaildation transform
self.config.dataset_params["diffusion_batch_size_valid"] = diffusion_batch_size
self.config.af3_inference["num_steps"] = num_steps