update to use retvec-v1 by default

google-research · Aug 2, 2023 · fcb6810 · fcb6810
1 parent 0d23517
commit fcb6810
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/retvec/tf/layers/tokenizer.py b/retvec/tf/layers/tokenizer.py
@@ -57,7 +57,7 @@ class RETVecTokenizer(tf.keras.layers.Layer):
     def __init__(
         self,
         sequence_length: int = 128,
-        model: Optional[Union[str, Path]] = None,
+        model: Optional[Union[str, Path]] = 'retvec-v1',
         trainable: bool = False,
         sep: str = "",
         standardize: Optional[str] = None,