Upload JudgeXL

judge_xl_model.py

@@ -28,9 +28,10 @@ class JudgeXLConfig(PretrainedConfig):
 class CustomEmbedding(nn.Module):
     def __init__(self, vocab_size, hidden_size):
         super(CustomEmbedding, self).__init__()
-        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}")
+        print(f"vocab_size: {vocab_size}, hidden_size: {hidden_size}")  # Debugging print
+        assert isinstance(vocab_size, int) and isinstance(hidden_size, int), \
+            f"Expected integers, but got vocab_size={type(vocab_size)} and hidden_size={type(hidden_size)}"
+        self.embedding = nn.Embedding(vocab_size, 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:7748f30578453658d57e62715496ea224ec2f5ed4d7a3c3134b509ecc9b895f4
+oid sha256:ba84e4ba48b14738177199bd52fcc4e25b376f8ce7aa5f91d9d7a304bf3d9b78
 size 1053691936