Data-Gouv-ML
/

RNE

PhysiQuanty commited on 17 days ago

Commit

adc8e4b

verified ·

1 Parent(s): 063d936

Update config.json

Files changed (1) hide show

config.json CHANGED Viewed

@@ -21,7 +21,7 @@
   "sep_token_id": 3,
   "pooling": "mean",
   "normalize_embeddings": true,
-  "attention_backend": "sage",
   "torch_fallback": false,
   "torch_dtype": "bfloat16",
   "transformers_version": "custom"

   "sep_token_id": 3,
   "pooling": "mean",
   "normalize_embeddings": true,
+  "attention_backend": "torch",
   "torch_fallback": false,
   "torch_dtype": "bfloat16",
   "transformers_version": "custom"