rio11user commited on
Commit
75963c6
·
verified ·
1 Parent(s): 0a8fa33

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. model.safetensors +1 -1
  2. modeling_simcse.py +2 -2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:28ad236e55d68accaeee647648beeb6ac0c7358fb885aa03cdbb91a459cf373e
3
  size 894432952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dcc80521e563196b22d864fceef8cc59ee2835b033bd0b7bea4a40531846830a
3
  size 894432952
modeling_simcse.py CHANGED
@@ -14,8 +14,8 @@ class SimCSEInferenceModel(PreTrainedModel):
14
  super().__init__(config)
15
  # 追加ダウンロードを避けるため from_config で空モデルを組み立てる
16
  base_cfg = BertConfig(**config.to_dict())
17
- self.encoder_input = BertModel.(base_cfg)
18
- self.encoder_output = BertModel.(base_cfg)
19
 
20
  hidden = self.encoder_input.config.hidden_size
21
  self.dense_input = nn.Linear(hidden, hidden)
 
14
  super().__init__(config)
15
  # 追加ダウンロードを避けるため from_config で空モデルを組み立てる
16
  base_cfg = BertConfig(**config.to_dict())
17
+ self.encoder_input = BertModel(base_cfg)
18
+ self.encoder_output = BertModel(base_cfg)
19
 
20
  hidden = self.encoder_input.config.hidden_size
21
  self.dense_input = nn.Linear(hidden, hidden)