From 511c4f93b065d604669edb9a1a74bfcfd72a1036 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 15 Oct 2024 10:39:25 -0700 Subject: [PATCH] bait --- vector_quantize_pytorch/lookup_free_quantization.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vector_quantize_pytorch/lookup_free_quantization.py b/vector_quantize_pytorch/lookup_free_quantization.py index 3d0e1d2..c8ff2e8 100644 --- a/vector_quantize_pytorch/lookup_free_quantization.py +++ b/vector_quantize_pytorch/lookup_free_quantization.py @@ -161,6 +161,7 @@ def __init__( self.activation = straight_through_activation + assert not (scale_trick and spherical) self.scale_trick = scale_trick # whether to use BSQ (binary spherical quantization)