File size: 327 Bytes
267e820 ef759d5 179c103 ef759d5 179c103 |
1 2 3 4 5 6 7 8 9 10 |
#!/usr/bin/env python3
from diffusers import DiffusionPipeline
from time import time
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=True, variant=v)
print(time() - start_time)
|