| # Which models to benchmark | |
| BENCHMARK_FUSONPLM = True | |
| # FUSONPLM_CKPTS. If you've traiend your own model, this is a dictionary: key = run name, values = epochs | |
| # If you want to use the trained FusOn-pLM, instead FUSONPLM_CKPTS="FusOn-pLM" | |
| FUSONPLM_CKPTS= "FusOn-pLM" | |
| BENCHMARK_ESM = True | |
| # GPU configs | |
| CUDA_VISIBLE_DEVICES="0" | |
| # Overwriting configs | |
| PERMISSION_TO_OVERWRITE_EMBEDDINGS = False # if False, script will halt if it believes these embeddings have already been made. | |
| PERMISSION_TO_OVERWRITE_MODELS = False # if False, script will halt if it believes these embeddings have already been made. | |