Update open_ai_image.py

This commit is contained in:
Haikui Yang
2023-12-17 19:50:06 +08:00
committed by GitHub
parent 04943c0bfa
commit 76a95c0226

View File

@@ -12,6 +12,7 @@ from config import conf
class OpenAIImage(object):
def __init__(self):
openai.api_key = conf().get("open_ai_api_key")
openai.api_base = conf().get("open_ai_api_base")
if conf().get("rate_limit_dalle"):
self.tb4dalle = TokenBucket(conf().get("rate_limit_dalle", 50))