From a10a476f124c55b714fd3aba06fd226e87ab8b17 Mon Sep 17 00:00:00 2001
From: Michael Shvartsman <mshvartsman@users.noreply.github.com>
Date: Tue, 13 Aug 2024 21:40:29 -0700
Subject: [PATCH] actually store the arg (#743)

---
 src/fairseq2/models/wav2vec2/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/fairseq2/models/wav2vec2/model.py b/src/fairseq2/models/wav2vec2/model.py
index 1f6dbc68d..d2f811fc5 100644
--- 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:
         """