Skip to content

Commit 3be26cf

Browse files
committed
add sd ut
1 parent af26fe0 commit 3be26cf

File tree

2 files changed

+26
-0
lines changed

2 files changed

+26
-0
lines changed

.gitignore

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,3 +178,12 @@ htmlcov.*
178178
*-structure.csv
179179
*-structure.json
180180

181+
/Titanic/2023_12_07_11_44_319a116fff/LLM_inout_pair/*.json
182+
/ICR/2023_12_06_14_14_26e593d09f/LLM_inout_pair/*.json
183+
/ICR/5cd9acb669c443fabe763e8f1ade5e86/workspace/*.txt
184+
/ICR/5cd9acb669c443fabe763e8f1ade5e86/workspace/*.csv
185+
/Titanic/9530b3c5550a4366ae92e5af6a74e6c3/workspace/*.csv
186+
/Titanic/9530b3c5550a4366ae92e5af6a74e6c3/workspace/*.txt
187+
/metagpt/roles/catboost_info/*.tsv
188+
/metagpt/roles/catboost_info/*.json
189+
/Titanic/9530b3c5550a4366ae92e5af6a74e6c3/workspace/*.md
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
# -*- coding: utf-8 -*-
2+
# @Date : 1/10/2024 10:07 PM
3+
# @Author : stellahong (stellahong@fuzhi.ai)
4+
# @Desc :
5+
from metagpt.tools.sd_engine import SDEngine
6+
7+
def test_sd_tools():
8+
engine = SDEngine()
9+
prompt = "1boy, hansom"
10+
engine.construct_payload(prompt)
11+
engine.simple_run_t2i(engine.payload)
12+
13+
def test_sd_construct_payload():
14+
engine = SDEngine()
15+
prompt = "1boy, hansom"
16+
engine.construct_payload(prompt)
17+
assert "negative_prompt" in engine.payload

0 commit comments

Comments
 (0)