apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

rm config get in dev

Browse source
This commit is contained in:
stellahsr 2024-01-31 16:30:50 +08:00
parent 28b0323d75
commit c44d08ceb0
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/tools/libs/sd_engine.py
View file

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