mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-16 01:14:47 +08:00
30 lines
962 B
Python
30 lines
962 B
Python
import argparse
|
|
import sys
|
|
|
|
|
|
sys.path.append(".")
|
|
from base_classes import ImageToImageBenchmark, TurboImageToImageBenchmark # noqa: E402
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--ckpt",
|
|
type=str,
|
|
default="Lykon/DreamShaper",
|
|
choices=[
|
|
"Lykon/DreamShaper",
|
|
"stabilityai/stable-diffusion-2-1",
|
|
"stabilityai/stable-diffusion-xl-refiner-1.0",
|
|
"stabilityai/sdxl-turbo",
|
|
],
|
|
)
|
|
parser.add_argument("--batch_size", type=int, default=1)
|
|
parser.add_argument("--num_inference_steps", type=int, default=50)
|
|
parser.add_argument("--model_cpu_offload", action="store_true")
|
|
parser.add_argument("--run_compile", action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
benchmark_pipe = ImageToImageBenchmark(args) if "turbo" not in args.ckpt else TurboImageToImageBenchmark(args)
|
|
benchmark_pipe.benchmark(args)
|