use torch.tensor(), not torch.Tensor()

huggingface · Nov 14, 2024 · 40e26ae · 40e26ae
1 parent a3d69a8
commit 40e26ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/mimi/modeling_mimi.py b/src/transformers/models/mimi/modeling_mimi.py
@@ -1214,7 +1214,7 @@ def __init__(self, config: MimiConfig, epsilon: float = 1e-5):
 
         self.codebook_size = config.codebook_size
 
-        self.register_buffer("initialized", torch.Tensor([True]))
+        self.register_buffer("initialized", torch.tensor([True]))
         self.register_buffer("cluster_usage", torch.ones(config.codebook_size))
         self.register_buffer("embed_sum", embed)
         self._embed = None