Update modeling.py

modeling.py

@@ -3,6 +3,14 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from dataclasses import dataclass
+from rwkv.model import RWKV
+
+class UpdatableRWKV(RWKV):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+    
+    def update_emb_weight(self, new_value):
+        self.w["emb.weight"] = new_value  
 
 @dataclass
 class VisualEncoderConfig: