model filename is configurable when loading pretrained

Author: lucidrains

alphafold3_pytorch/alphafold3.py

@@ -3652,9 +3652,9 @@ def _from_pretrained(
         token: str | bool | None,
         map_location: str = 'cpu',
         strict: bool = False,
+        model_filename: str = 'alphafold3.bin',
         **model_kwargs,
     ):
-        model_filename = "alphafold3.bin"
         model_file = Path(model_id) / model_filename
 
         if not model_file.exists():