code: def forward(self, x):
      return torch.from_numpy(
          self.kmeans_model.predict(x.cpu().numpy())
      ).to(self.device)
  
file_path: textlesslib/textless/data/kmeans_quantizer.py
imports: import torch
  import joblib
  import warnings
label_desc: Use image node_id 2 for calling the KMeansQuantizer obj's underlying member method code with example usage: obj.forward(x) and returns: torch