clean up some args'

This commit is contained in:
ita.zaporozhets@huggingface.co 2025-06-23 13:56:30 +00:00 committed by ita.zaporozhets@huggingface.co
parent 91be87ec9b
commit 2f23e3efd5

View File

@ -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