From c44d08ceb05ee177915506a84fc40b021ef4698c Mon Sep 17 00:00:00 2001 From: stellahsr Date: Wed, 31 Jan 2024 16:30:50 +0800 Subject: [PATCH] rm config get in dev --- metagpt/tools/libs/sd_engine.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/metagpt/tools/libs/sd_engine.py b/metagpt/tools/libs/sd_engine.py index 57a025f3c..7001eadf5 100644 --- a/metagpt/tools/libs/sd_engine.py +++ b/metagpt/tools/libs/sd_engine.py @@ -56,11 +56,9 @@ default_negative_prompt = "(easynegative:0.8),black, dark,Low resolution" @register_tool(tool_type=ToolTypeEnum.STABLE_DIFFUSION.value) class SDEngine: def __init__(self, sd_url=""): - from metagpt.config2 import config - # Initialize the SDEngine with configuration - self.sd_url = sd_url if sd_url else config.get("SD_URL") - self.sd_t2i_url = f"{self.sd_url}{config.get('SD_T2I_API')}" + self.sd_url = sd_url + self.sd_t2i_url = f"{self.sd_url}/sdapi/v1/txt2img" # Define default payload settings for SD API self.payload = payload logger.info(self.sd_t2i_url)