Hugging Face's logo

Omaratef3221
/
custom_bert_with_rbf

Safetensors
bert
Model card Files Community
Omaratef3221 commited on
Commit
df76e5f
·
verified ·
1 Parent(s): 7563185

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. CustomBertModel.py +1 -1
  2. model.safetensors +1 -1
CustomBertModel.py CHANGED
@@ -3,7 +3,7 @@ import torch.nn as nn
3
  from RBFLayer import RBFLayer # Import your RBF implementation
4
  import torch
5
 
6
- class CustomBertModel(BertModel):
7
  def __init__(self, config):
8
  super().__init__(config)
9
 
 
3
  from RBFLayer import RBFLayer # Import your RBF implementation
4
  import torch
5
 
6
+ class CustomBertModel(AutoModel):
7
  def __init__(self, config):
8
  super().__init__(config)
9
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4300ef8324f02981ea1fe0dd6246bff23c2d4d29ab4b79fd8af6b0d04b0dce50
3
  size 212015704
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb4a2d26ca5b8400009e886c555c1a517c0801156e83d09fd21692c4dc6d555c
3
  size 212015704