johahi commited on
Commit
f7a738d
·
verified ·
1 Parent(s): 17f726c

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
README.md ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ After the installation of FlashAttention2, the model can be loaded using:
2
+ ```python
3
+ from scooby.modeling import Scooby
4
+ flooby_neurips = Scooby.from_pretrained(
5
+ 'johahi/flash-scooby-neurips',
6
+ cell_emb_dim=14,
7
+ embedding_dim=1920,
8
+ n_tracks=3,
9
+ return_center_bins_only=True,
10
+ disable_cache=False,
11
+ use_transform_borzoi_emb=False,
12
+ )
13
+ ```
14
+ 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.