examples/alphafold3.md
... ...
@@ -99,6 +99,7 @@ input_json=${PROJECT_DIR}/alphafold_input.json
99 99
100 100
time run_alphafold.py \
101 101
--json_path=${input_json} \
102
+ --flash_attention_implementation=xla \
102 103
--output_dir=${PROJECT_DIR}/af3_example_output \
103 104
--db_dir=${DATABASE_DIR} \
104 105
--model_dir=${MODEL_DIR}