File size: 370 Bytes
d1f1095 | 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)
|