init commit

test.py

@@ -0,0 +1,13 @@
+from diffusers import DiffusionPipeline
+import torch
+import time
+
+model_path = "/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5"
+# model_path = "/mnt/contest_ceph/zhanghao/models/stable-diffusion-3.5-medium"
+pipeline = DiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
+pipeline.to("cuda")
+start = time.time()
+image = pipeline("An image of a squirrel in Picasso style").images[0]
+end = time.time()
+print(f"elapsed: {end - start}")
+image.save("squirrel_picasso.png")