diff --git a/scripts/hook.py b/scripts/hook.py index 767a7db09..f43b34acd 100644 --- a/scripts/hook.py +++ b/scripts/hook.py @@ -28,7 +28,7 @@ BasicTransformerBlockSGM = BasicTransformerBlock -POSITIVE_MARK_TOKEN = 1024 +POSITIVE_MARK_TOKEN = 1000 NEGATIVE_MARK_TOKEN = - POSITIVE_MARK_TOKEN MARK_EPS = 1e-3