elijahcole Saarthak-GenBio-AI commited on
Commit
bd4dc4b
·
1 Parent(s): fb58b27

Create configuration_genbio_pathfm.py (#1)

Browse files

- Create configuration_genbio_pathfm.py (4af56e2c0926f5b249dc52271df48f96c79f16eb)


Co-authored-by: Saarthak Kapse <Saarthak-GenBio-AI@users.noreply.huggingface.co>

Files changed (1) hide show
  1. configuration_genbio_pathfm.py +63 -0
configuration_genbio_pathfm.py ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """GenBio-PathFM configuration for HuggingFace AutoModel."""
2
+
3
+ from transformers import PretrainedConfig
4
+
5
+
6
+ class GenBioPathFMConfig(PretrainedConfig):
7
+ model_type = "genbio_pathfm"
8
+
9
+ def __init__(
10
+ self,
11
+ img_size=224,
12
+ patch_size=16,
13
+ in_chans=1,
14
+ embed_dim=1536,
15
+ depth=40,
16
+ num_heads=24,
17
+ ffn_ratio=4,
18
+ n_storage_tokens=4,
19
+ ffn_layer="swiglu64",
20
+ layerscale_init=1e-5,
21
+ qkv_bias=False,
22
+ proj_bias=True,
23
+ ffn_bias=True,
24
+ norm_layer="layernorm",
25
+ drop_path_rate=0.0,
26
+ pos_embed_rope_base=100.0,
27
+ pos_embed_rope_min_period=None,
28
+ pos_embed_rope_max_period=None,
29
+ pos_embed_rope_normalize_coords="separate",
30
+ pos_embed_rope_shift_coords=None,
31
+ pos_embed_rope_jitter_coords=True,
32
+ pos_embed_rope_rescale_coords=2,
33
+ pos_embed_rope_dtype="bf16",
34
+ image_mean=(0.697, 0.575, 0.728),
35
+ image_std=(0.188, 0.240, 0.187),
36
+ **kwargs,
37
+ ):
38
+ super().__init__(**kwargs)
39
+ self.img_size = img_size
40
+ self.patch_size = patch_size
41
+ self.in_chans = in_chans
42
+ self.embed_dim = embed_dim
43
+ self.depth = depth
44
+ self.num_heads = num_heads
45
+ self.ffn_ratio = ffn_ratio
46
+ self.n_storage_tokens = n_storage_tokens
47
+ self.ffn_layer = ffn_layer
48
+ self.layerscale_init = layerscale_init
49
+ self.qkv_bias = qkv_bias
50
+ self.proj_bias = proj_bias
51
+ self.ffn_bias = ffn_bias
52
+ self.norm_layer = norm_layer
53
+ self.drop_path_rate = drop_path_rate
54
+ self.pos_embed_rope_base = pos_embed_rope_base
55
+ self.pos_embed_rope_min_period = pos_embed_rope_min_period
56
+ self.pos_embed_rope_max_period = pos_embed_rope_max_period
57
+ self.pos_embed_rope_normalize_coords = pos_embed_rope_normalize_coords
58
+ self.pos_embed_rope_shift_coords = pos_embed_rope_shift_coords
59
+ self.pos_embed_rope_jitter_coords = pos_embed_rope_jitter_coords
60
+ self.pos_embed_rope_rescale_coords = pos_embed_rope_rescale_coords
61
+ self.pos_embed_rope_dtype = pos_embed_rope_dtype
62
+ self.image_mean = image_mean
63
+ self.image_std = image_std