Commit 1cddb60b authored by lijian6's avatar lijian6
Browse files

Initial commit


Signed-off-by: lijian6's avatarlijian <lijian6@sugon.com>
parents
[submodule "diffusers"]
path = diffusers
url = http://developer.hpccube.com/codes/OpenDAS/diffusers.git
import torch
from diffusers import HunyuanDiTPipeline
import time
pipe = HunyuanDiTPipeline.from_pretrained("HunyuanDiT-Diffusers/", torch_dtype=torch.float16)
pipe.to("cuda")
base_count = 0
prompt = "一个宇航员在骑马"
for i in range(2):
start = time.time()
image = pipe(prompt).images[0]
print(f"the {i} text-to-image use time {time.time()-start}")
base_count += 1
image.save(f"Hunyuan-{i:05}.png")
diffusers @ a9ae9148
Subproject commit a9ae9148cc20f383cb95c6bf384fc83be3b81005
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment