diff --git a/nodes.py b/nodes.py index 923d294..cfdf837 100644 --- a/nodes.py +++ b/nodes.py @@ -36,6 +36,7 @@ try: from accelerate.utils import set_module_tensor_to_device is_accelerate_available = True except: + is_accelerate_available = False pass