mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 18:51:14 +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__()
|
super().__init__()
|
||||||
|
|
||||||
# Extract parameters from dictionary
|
# Extract parameters from dictionary
|
||||||
dim = dim
|
|
||||||
n_heads = n_heads
|
|
||||||
head_dim = config.head_dim
|
head_dim = config.head_dim
|
||||||
n_kv_heads = config.n_kv_heads
|
n_kv_heads = config.n_kv_heads
|
||||||
rope_theta = config.rope_theta
|
|
||||||
multiple_of = config.multiple_of
|
multiple_of = config.multiple_of
|
||||||
ffn_dim_multiplier = config.ffn_dim_multiplier
|
|
||||||
norm_eps = config.norm_eps
|
norm_eps = config.norm_eps
|
||||||
|
|
||||||
self.head_dim = head_dim or dim // n_heads
|
self.head_dim = head_dim or dim // n_heads
|
||||||
|
Loading…
Reference in New Issue
Block a user