wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 21:42:37 +08:00
Code Issues Actions 13 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": { "13": {
"inputs": { "inputs": {
"filename_prefix": "tests/ref_ComfyUI", "filename_prefix": "test_ComfyUI",
"images": [ "images": [
"12", "12",
0 0

8
tests/inference/test_inference.py
View File

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