File size: 557 Bytes
37f5221
 
 
 
 
 
 
f7a738d
 
 
 
06bbc89
f7a738d
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
---
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.