ThingsAI commited on
Commit
ffe51ed
·
verified ·
1 Parent(s): 081a4b9

Upload configuration_quark.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. configuration_quark.py +12 -0
configuration_quark.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import PretrainedConfig
2
+
3
+ class QuarkConfig(PretrainedConfig):
4
+ model_type = "quark"
5
+ def __init__(self, vocab_size=65537, d_model=768, n_heads=12, n_kv_heads=4,
6
+ n_layers=32, d_ff=2048, head_dim=64, max_seq_len=2048,
7
+ rope_theta=10000.0, rms_eps=1e-5, qkv_bias=True, dropout=0.0, **kwargs):
8
+ self.vocab_size=vocab_size; self.d_model=d_model; self.n_heads=n_heads
9
+ self.n_kv_heads=n_kv_heads; self.n_layers=n_layers; self.d_ff=d_ff
10
+ self.head_dim=head_dim; self.max_seq_len=max_seq_len; self.rope_theta=rope_theta
11
+ self.rms_eps=rms_eps; self.qkv_bias=qkv_bias; self.dropout=dropout
12
+ super().__init__(**kwargs)