jonathan-lys commited on
Commit
befa5b5
·
verified ·
1 Parent(s): c6d1fa7

Update position_bank.py

Browse files
Files changed (1) hide show
  1. position_bank.py +1 -0
position_bank.py CHANGED
@@ -13,6 +13,7 @@ class RevePositionBank(PreTrainedModel):
13
  self.position_names = config.position_names
14
  self.mapping = {name: i for i, name in enumerate(self.position_names)}
15
  self.register_buffer("embedding", torch.randn(len(self.position_names), 3))
 
16
 
17
  def forward(self, channel_names: list[str]):
18
  indices = [self.mapping[q] for q in channel_names if q in self.mapping]
 
13
  self.position_names = config.position_names
14
  self.mapping = {name: i for i, name in enumerate(self.position_names)}
15
  self.register_buffer("embedding", torch.randn(len(self.position_names), 3))
16
+ self.post_init()
17
 
18
  def forward(self, channel_names: list[str]):
19
  indices = [self.mapping[q] for q in channel_names if q in self.mapping]