                       self.__class__.__name__, name, model_path, model_kwargs, allow_growth,
                       exclude_gpus)
          self._name = name
          self._backend = get_backend()
          self._set_session(allow_growth, [] if exclude_gpus is None else exclude_gpus)
          self._model_path = model_path
          self._model_kwargs = {} if not model_kwargs else model_kwargs
          self._model: Optional[Model] = None
          logger.trace("Initialized: %s", self.__class__.__name__,)  # type:ignore
  
 
</s> remove                      exclude_gpus)
 </s> add                      exclude_gpus, cpu_mode) </s> remove                      "model_kwargs: %s,  allow_growth: %s, exclude_gpus: %s)",
 </s> add                      "model_kwargs: %s,  allow_growth: %s, exclude_gpus: %s, cpu_mode: %s)", </s> remove                  exclude_gpus: Optional[List[int]] = None) -> None:
 </s> add                  exclude_gpus: Optional[List[int]] = None,
                 cpu_mode: bool = False) -> None: </s> remove         self.vram = 320
        self.vram_warnings = 64  # Will run at this with warnings
        self.vram_per_batch = 32
 </s> add         self.vram = 320 if not self.config["cpu"] else 0
        self.vram_warnings = 64 if not self.config["cpu"] else 0  # Will run at this with warnings
        self.vram_per_batch = 32 if not self.config["cpu"] else 0 </s> remove         self._model.load_weights(self._model_path)
        if self._backend != "amd":
            self._model.make_predict_function()
 </s> add         with self._context:
            self._model.load_weights(self._model_path)
            if self._backend != "amd":
                self._model.make_predict_function()
        self._context = self._set_session(allow_growth,
                                          [] if exclude_gpus is None else exclude_gpus,
                                          cpu_mode)
                       self.__class__.__name__, name, model_path, model_kwargs, allow_growth,
                       exclude_gpus)
          self._name = name
          self._backend = get_backend()
        self._context = self._set_session(allow_growth,
                                          [] if exclude_gpus is None else exclude_gpus,
                                          cpu_mode)
          self._model_path = model_path
          self._model_kwargs = {} if not model_kwargs else model_kwargs
          self._model: Optional[Model] = None
          logger.trace("Initialized: %s", self.__class__.__name__,)  # type:ignore
  
keep keep keep keep replace keep keep keep keep keep
 Add CPU option to BiSeNet and MTCNN
  - Add CPU option to KSession
  - MTCNN optimizations
  - Update docs for bisenet + mtcnn 
lib/model/session.py