Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
RuiWang1998 committed Aug 13, 2022
1 parent 1fd869f commit 7df5b12
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,7 @@ releasing possibly stronger models.
We have optimized (to some extent) the GRAM usage of OmegaFold model in our
latest release. Now the model can inference protein sequence as long as
_4096_ on NVIDIA A100 Graphics card with 80 GB of memory with
`--subbatch_size` set to 448 without hitting 70 GB of memory.
You could also inference a 4096-long sequence with `--subbatch_size` set to
1 on a 24 GB GPU (but might take _a while_).
`--subbatch_size` set to 448 without hitting full memory.
This version's model is more sensitive to `--subbatch_size`.

### Setting Subbatch
Expand All @@ -30,9 +28,9 @@ but we suggest half the value if you run into GPU memory limitations.

### MacOS Users

For macOS users, we support MPS (Apple Silicon) acceleration if the user
For macOS users, we support MPS (Apple Silicon) acceleration if the user
installs the latest nightly version of PyTorch.
Also, current code also requires macOS users need to `git clone` the
Also, current code also requires macOS users need to `git clone` the
repository and use `python main.
py` (see below) to run the model.

Expand Down

0 comments on commit 7df5b12

Please sign in to comment.