class_docstr: Wrapper object for a BlockData which similarity search via FAISS under the hood
class_docstr_tok: ["Wrapper","object","for","a","BlockData","which","similarity","search","via","FAISS","under","the","hood"]
code: def __init__(self, embed_size, embed_data=None, use_gpu=False):
      self.embed_size = embed_size
      self.embed_data = embed_data
      self.use_gpu = use_gpu
  
      self.mips_index = None
      self._set_mips_index()
  
imports: import itertools
  import os
  import pickle
  import shutil
  import numpy
  import torch
  from megatron import get_args
  from megatron.core import mpu
label_desc: Use image node_id 1 to create a new FaissMIPSIndex object from inherited base classes: object with example: obj = FaissMIPSIndex(embed_size, embed_data, use_gpu)