From bc0828535605963b06bbd99a09aef991fa12e9c5 Mon Sep 17 00:00:00 2001
From: qutcat1997 <54978665+qutcat1997@users.noreply.github.com>
Date: Tue, 11 Jul 2023 23:06:41 +0800
Subject: [PATCH] Update config.py

Modify config
---
 pilot/configs/config.py | 2 ++
 1 file changed, 2 insertions(+)

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"
         )