  import numpy as np
  
  from lib.multithreading import MultiThread
  from lib.queue_manager import queue_manager
  from lib.utils import GetModel, FaceswapError
  from ._config import Config
  
 
</s> remove from lib.utils import GetModel
 </s> add from lib.utils import GetModel, FaceswapError </s> remove             batch = function(batch)
 </s> add             try:
                batch = function(batch)
            except tf_errors.UnknownError as err:
                msg = ("Tensorflow raised an unknown error. This is most likely caused by a "
                       "failure to launch cuDNN which can occur for some GPU/Tensorflow "
                       "combinations. You should enable `allow_growth` to attempt to resolve this "
                       "issue:"
                       "\nGUI: Go to Settings > Extract Plugins > Global and enable the "
                       "`allow_growth` option."
                       "\nCLI: Go to `faceswap/config/extract.ini` and change the `allow_growth "
                       "option to `True`.")
                raise FaceswapError(msg) from err
from tensorflow.python import errors_impl as tf_errors  # pylint:disable=no-name-in-module

  import numpy as np
  
from tensorflow.python import errors_impl as tf_errors  # pylint:disable=no-name-in-module

  from lib.multithreading import MultiThread
  from lib.queue_manager import queue_manager
  from lib.utils import GetModel, FaceswapError
  from ._config import Config
  
keep add keep keep keep keep keep
 Capture cuDNN error in extract and raise a useful message 
plugins/extract/_base.py