mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
clean up some args'
This commit is contained in:
parent
91be87ec9b
commit
2f23e3efd5
@ -238,13 +238,9 @@ class BLTTransformerLayer(nn.Module):
|
||||
super().__init__()
|
||||
|
||||
# Extract parameters from dictionary
|
||||
dim = dim
|
||||
n_heads = n_heads
|
||||
head_dim = config.head_dim
|
||||
n_kv_heads = config.n_kv_heads
|
||||
rope_theta = config.rope_theta
|
||||
multiple_of = config.multiple_of
|
||||
ffn_dim_multiplier = config.ffn_dim_multiplier
|
||||
norm_eps = config.norm_eps
|
||||
|
||||
self.head_dim = head_dim or dim // n_heads
|
||||
|
Loading…
Reference in New Issue
Block a user