diff --git a/pilot/configs/config.py b/pilot/configs/config.py index 804176357..1c8026df5 100644 --- a/pilot/configs/config.py +++ b/pilot/configs/config.py @@ -28,6 +28,8 @@ class Config(metaclass=Singleton): self.skip_reprompt = False self.temperature = float(os.getenv("TEMPERATURE", 0.7)) + self.NUM_GPUS = int(os.getenv("NUM_GPUS",1)) + self.execute_local_commands = ( os.getenv("EXECUTE_LOCAL_COMMANDS", "False") == "True" )