qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Unhardcode sampler and scheduler list in test.

This commit is contained in:
comfyanonymous 2023-09-18 23:23:25 -04:00
parent 7c93afd2cd
commit f32463936d
1 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/inference/test_inference.py
View File

@ -16,10 +16,8 @@ import uuid
import urllib.request
import urllib.parse
# Currently causes an error when running pytest with built-in pytest args
# TODO: modify cli_args.py to not parse args on import
# We will hard-code sampler and scheduler lists for now
# from comfy.samplers import KSampler
from comfy.samplers import KSampler
"""
These tests generate and save images through a range of parameters
@ -140,16 +138,10 @@ comfy_graph_ids = [DEFAULT_COMFY_GRAPH_ID]
prompt_list = [
'a painting of a cat',
]
#TODO use sampler and scheduler list from comfy.samplers.KSampler
# sampler_list = KSampler.SAMPLERS
# scheduler_list = KSampler.SCHEDULERS
# Hard coded sampler and scheduler lists for now
SCHEDULERS = ["normal", "karras", "exponential", "sgm_uniform", "simple", "ddim_uniform"]
SAMPLERS = ["euler", "euler_ancestral", "heun", "dpm_2", "dpm_2_ancestral",
"lms", "dpm_fast", "dpm_adaptive", "dpmpp_2s_ancestral", "dpmpp_sde", "dpmpp_sde_gpu",
"dpmpp_2m", "dpmpp_2m_sde", "dpmpp_2m_sde_gpu", "dpmpp_3m_sde", "dpmpp_3m_sde_gpu", "ddim", "uni_pc", "uni_pc_bh2"]
sampler_list = SAMPLERS
scheduler_list = SCHEDULERS
sampler_list = KSampler.SAMPLERS
scheduler_list = KSampler.SCHEDULERS
@pytest.mark.inference
@pytest.mark.parametrize("sampler", sampler_list)
@pytest.mark.parametrize("scheduler", scheduler_list)