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