sentence-transformers
/

all-mpnet-base-v2

When embedding in CUDA-enable environment, the Device side Assertion error occur when I try to put more than 512 embedding dimension. It looks like the config.json is wrong, the real max_position_embeddings are 512.

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -12,7 +12,7 @@
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "layer_norm_eps": 1e-05,
-  "max_position_embeddings": 514,
   "model_type": "mpnet",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,

   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "layer_norm_eps": 1e-05,
+  "max_position_embeddings": 512,
   "model_type": "mpnet",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,