19 lines
588 B
Python
19 lines
588 B
Python
import logging
|
|
from rich.logging import RichHandler
|
|
from modules.shared import cmd_opts
|
|
|
|
log_level = 'DEBUG' if cmd_opts.distributed_debug else 'INFO'
|
|
logger = logging.getLogger("rich")
|
|
logger.addHandler(RichHandler(rich_tracebacks=True, markup=True, show_time=False))
|
|
logger.setLevel(log_level)
|
|
|
|
warmup_samples = 2 # number of samples to do before recording a valid benchmark sample
|
|
benchmark_payload: dict = {
|
|
"prompt": "A herd of cows grazing at the bottom of a sunny valley",
|
|
"negative_prompt": "",
|
|
"steps": 20,
|
|
"width": 512,
|
|
"height": 512,
|
|
"batch_size": 1
|
|
}
|