Update config.py

Modify config
This commit is contained in:
qutcat1997 2023-07-11 23:06:41 +08:00 committed by GitHub
parent 65be62c2b0
commit bc08285356
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -28,6 +28,8 @@ class Config(metaclass=Singleton):
self.skip_reprompt = False self.skip_reprompt = False
self.temperature = float(os.getenv("TEMPERATURE", 0.7)) self.temperature = float(os.getenv("TEMPERATURE", 0.7))
self.NUM_GPUS = int(os.getenv("NUM_GPUS",1))
self.execute_local_commands = ( self.execute_local_commands = (
os.getenv("EXECUTE_LOCAL_COMMANDS", "False") == "True" os.getenv("EXECUTE_LOCAL_COMMANDS", "False") == "True"
) )