File size: 368 Bytes
267e820 ef759d5 179c103 ef759d5 a89aa7e |
1 2 3 4 5 6 7 8 9 10 11 |
#!/usr/bin/env python3
from diffusers import DiffusionPipeline
from time import time
for y in [True, False]:
for v in ["no_ema", "fp16", None]:
start_time = time()
p = DiffusionPipeline.from_pretrained("huggingface/the-no-branch-repo", variant=v)
p.save_pretrained("./", safe_serialization=y, variant=v)
print(time() - start_time)
|