howard-hou commited on
Commit
c71bb52
·
1 Parent(s): 3c371fd

Update modeling.py

Browse files
Files changed (1) hide show
  1. modeling.py +1 -1
modeling.py CHANGED
@@ -10,7 +10,7 @@ class UpdatableRWKV(RWKV):
10
  super().__init__(*args, **kwargs)
11
 
12
  def update_emb_weight(self, new_value):
13
- self.w["emb.weight"] = new_value
14
 
15
  @dataclass
16
  class VisualEncoderConfig:
 
10
  super().__init__(*args, **kwargs)
11
 
12
  def update_emb_weight(self, new_value):
13
+ self.w.update({"emb.weight", new_value})
14
 
15
  @dataclass
16
  class VisualEncoderConfig: