johahi's picture
Update README.md
06bbc89 verified
---
license: mit
tags:
- genomics
- single-cell
- biology
---
After the installation of FlashAttention2, the model can be loaded using:
```python
from scooby.modeling import Scooby
flooby_neurips = Scooby.from_pretrained(
'johahi/neurips-scooby-flash',
cell_emb_dim=14,
embedding_dim=1920,
n_tracks=3,
return_center_bins_only=True,
disable_cache=False,
use_transform_borzoi_emb=False,
)
```
Since the transform_borzoi_emb is not used, the models performance is slightly (0.52 vs 0.54 across cell types) worse than neurips-scooby.