diff --git a/components/caption_images/src/main.py b/components/caption_images/src/main.py index 3bfa4c922..9f6d85107 100644 --- a/components/caption_images/src/main.py +++ b/components/caption_images/src/main.py @@ -13,7 +13,6 @@ logger = logging.getLogger(__name__) -os.environ['CUDA_LAUNCH_BLOCKING'] = "1" os.environ['TORCH_CUDNN_V8_API_DISABLED'] = "1" diff --git a/components/embed_images/src/main.py b/components/embed_images/src/main.py index 72a43cc52..d08ed10b7 100644 --- a/components/embed_images/src/main.py +++ b/components/embed_images/src/main.py @@ -13,7 +13,6 @@ logger = logging.getLogger(__name__) -os.environ['CUDA_LAUNCH_BLOCKING'] = "1" os.environ['TORCH_CUDNN_V8_API_DISABLED'] = "1" diff --git a/components/segment_images/src/main.py b/components/segment_images/src/main.py index cf9bfaea0..3ce23acc4 100644 --- a/components/segment_images/src/main.py +++ b/components/segment_images/src/main.py @@ -14,7 +14,6 @@ logger = logging.getLogger(__name__) -os.environ['CUDA_LAUNCH_BLOCKING'] = "1" os.environ['TORCH_CUDNN_V8_API_DISABLED'] = "1" diff --git a/examples/pipelines/datacomp/components/detect_text/src/main.py b/examples/pipelines/datacomp/components/detect_text/src/main.py index 1b83492f4..a24c5e84e 100644 --- a/examples/pipelines/datacomp/components/detect_text/src/main.py +++ b/examples/pipelines/datacomp/components/detect_text/src/main.py @@ -23,7 +23,6 @@ logger = logging.getLogger(__name__) -os.environ["CUDA_LAUNCH_BLOCKING"] = "1" os.environ["TORCH_CUDNN_V8_API_DISABLED"] = "1"