diff --git a/extensions/openai/script.py b/extensions/openai/script.py index 58247917..f9373385 100644 --- a/extensions/openai/script.py +++ b/extensions/openai/script.py @@ -8,7 +8,7 @@ from modules import shared from modules.text_generation import encode, generate_reply params = { - 'port': int(os.environ('OPENEDAI_PORT')) if 'OPENEDAI_PORT' in os.environ else 5001, + 'port': int(os.environ.get('OPENEDAI_PORT')) if 'OPENEDAI_PORT' in os.environ else 5001, } debug = True if 'OPENEDAI_DEBUG' in os.environ else False