kaizen9 commited on
Commit
023a725
1 Parent(s): 7a7db37

Update modeling_mpt.py

Browse files
Files changed (1) hide show
  1. 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']