Upload JudgeXL

judge_xl_model.py

@@ -28,8 +28,10 @@ class JudgeXLConfig(PretrainedConfig):
 class CustomEmbedding(nn.Module):
     def __init__(self, vocab_size, hidden_size):
         super(CustomEmbedding, self).__init__()
-        self.embedding = nn.Embedding(vocab_size, hidden_size)
         assert isinstance(vocab_size, int) and isinstance(hidden_size, int)
+        self.embedding = nn.Embedding(config.vocab_size, config.hidden_size)
+        print(f"vocab_size: {config.vocab_size}, hidden_size: {config.hidden_size}")
+
     def forward(self, inputs):
         return self.embedding(inputs)
 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2b17fb87eb9a0731e250e91bc45ce5d2cdd943c44eb02775767add2b8a5d81da
+oid sha256:7748f30578453658d57e62715496ea224ec2f5ed4d7a3c3134b509ecc9b895f4
 size 1053691936