From cfb86cc293b6b3712aeca026f6ce1970cf56ca64 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Tue, 17 Dec 2024 16:39:22 -0800 Subject: [PATCH] Add clarifying comment from cuda_accelerator --- accelerator/cpu_accelerator.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/accelerator/cpu_accelerator.py b/accelerator/cpu_accelerator.py index 572191ff9c13..de711f73144e 100644 --- a/accelerator/cpu_accelerator.py +++ b/accelerator/cpu_accelerator.py @@ -5,6 +5,8 @@ from .abstract_accelerator import DeepSpeedAccelerator +# During setup stage torch may not be installed, pass on no torch will +# allow op builder related API to be executed. try: import torch except ImportError as e: