wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 13:32:36 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Clean up

Browse Source
This commit is contained in:
enzymezoo-code 2023-09-01 16:27:04 -05:00
parent 66725034de
commit cd18e72bf3
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/inference/graphs/default_graph.json
View File

@ -68,7 +68,7 @@
},
"13": {
"inputs": {
"filename_prefix": "tests/ref_ComfyUI",
"filename_prefix": "test_ComfyUI",
"images": [
"12",
0

8
tests/inference/test_inference.py
View File

@ -19,8 +19,7 @@ import urllib.parse
from comfy.samplers import KSampler
"""
These tests are used to generate images compare the output
of inference with the ground-truth images.
These tests are to generate and save images through a range of parameters
"""
class ComfyGraph:
@ -127,8 +126,8 @@ DEFAULT_COMFY_GRAPH = ComfyGraph(graph=default_graph, sampler_nodes=['10','14'])
#
comfy_graph_list = [DEFAULT_COMFY_GRAPH]
prompt_list = [
'a textured painting of a wet frog',
'a textured painting of a wet toad',
'a painting of a cat',
'a photo of a toad',
]
sampler_list = KSampler.SAMPLERS[0:2]
scheduler_list = [KSampler.SCHEDULERS[0]]
@ -200,5 +199,6 @@ class TestInference:
comfy_graph.set_scheduler(scheduler)
comfy_graph.set_prompt(prompt)
# Generate
client.get_images(comfy_graph.graph)