Merge branch 'main' into cli/tab-completion

Ahmedsaed · Aug 15, 2024 · 7708752 · 7708752
2 parents 9e91069 + a10a476
commit 7708752
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/fairseq2/models/wav2vec2/model.py b/src/fairseq2/models/wav2vec2/model.py
@@ -109,6 +109,7 @@ def __init__(
 
         self.num_distractors = num_distractors
         self.logit_temp = logit_temp
+        self.quantizer_encoder_grad = quantizer_encoder_grad
 
     def forward(self, batch: SequenceBatch) -> Wav2Vec2Output:
         """