Upload JudgeXL

judge_xl_model.py

@@ -29,7 +29,7 @@ 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)
     def forward(self, inputs):
         return self.embedding(inputs)
 

model.safetensors

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