fix the is_empty logic

Signed-off-by: Yu Chin Fabian Lim <[email protected]>

tuning/config/acceleration_configs/acceleration_framework_config.py

@@ -162,8 +162,6 @@ def from_dataclasses(*dataclasses: Type):
         return config
 
     def get_framework(self):
-        if self.is_empty():
-            return
 
         if is_fms_accelerate_available():
 
@@ -174,17 +172,31 @@ def get_framework(self):
                 NamedTemporaryFile,
             )
 
-            with NamedTemporaryFile("w") as f:
-                self.to_yaml(f.name)
-                return AccelerationFramework(f.name)
+            try:
+                with NamedTemporaryFile("w") as f:
+                    self.to_yaml(f.name)
+                    return AccelerationFramework(f.name)
+            except ValueError as e:
+                (msg,) = e.args
+
+                # AcceleratorFramework raises ValueError if it
+                # fails to configure any plugin
+                if self.is_empty() and msg.startswith("No plugins could be configured"):
+                    # in the case when the error was thrown when
+                    # the acceleration framework config was empty
+                    # then this is expected.
+                    return None
+
+                raise e
         else:
-            raise ValueError(
-                "No acceleration framework package found. To use, first "
-                "ensure that 'pip install fms-hf-tuning[fms-accel]' is done first to "
-                "obtain the acceleration framework dependency. Additional "
-                "acceleration plugins make be required depending on the requsted "
-                "acceleration. See README.md for instructions."
-            )
+            if not self.is_empty():
+                raise ValueError(
+                    "No acceleration framework package found. To use, first "
+                    "ensure that 'pip install fms-hf-tuning[fms-accel]' is done first to "
+                    "obtain the acceleration framework dependency. Additional "
+                    "acceleration plugins make be required depending on the requsted "
+                    "acceleration. See README.md for instructions."
+                )
 
     def is_empty(self):
         "check if the configuration is empty"