Update modeling_mpt.py
Browse files- modeling_mpt.py +2 -1
modeling_mpt.py
CHANGED
@@ -193,7 +193,8 @@ class MPTModel(MPTPreTrainedModel):
|
|
193 |
self.emb_drop = nn.Dropout(config.emb_pdrop)
|
194 |
self.blocks = nn.ModuleList([MPTBlock(device=config.init_device, **config.to_dict()) for _ in range(config.n_layers)])
|
195 |
self.norm_f = norm_class(config.d_model, device=config.init_device)
|
196 |
-
self.rope = config.attn_config['rope']
|
|
|
197 |
self.rope_impl = None
|
198 |
if self.rope:
|
199 |
self.rope_impl = config.attn_config['rope_impl']
|
|
|
193 |
self.emb_drop = nn.Dropout(config.emb_pdrop)
|
194 |
self.blocks = nn.ModuleList([MPTBlock(device=config.init_device, **config.to_dict()) for _ in range(config.n_layers)])
|
195 |
self.norm_f = norm_class(config.d_model, device=config.init_device)
|
196 |
+
#self.rope = config.attn_config['rope']
|
197 |
+
self.rope = False
|
198 |
self.rope_impl = None
|
199 |
if self.rope:
|
200 |
self.rope_impl = config.attn_config['rope_impl']
|