Tencent-Hunyuan
/

HunyuanDiT-v1.2-ControlNet-Diffusers-Pose

Model card Files Files and versions Community

Zhiminli commited on Jul 14

Commit

e39decc

•

1 Parent(s): 99d3010

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -9,7 +9,7 @@ from diffusers import HunyuanDiT2DControlNetModel, HunyuanDiTControlNetPipeline
 import torch
-controlnet = HunyuanDiT2DControlNetModel.from_pretrained("Tencent-Hunyuan/HunyuanDiT-v1.2-ControlNet-Diffusers-Distilled-Pose", torch_dtype=torch.float16)
 pipe = HunyuanDiTControlNetPipeline.from_pretrained("Tencent-Hunyuan/HunyuanDiT-v1.2-Diffusers-Distilled", controlnet=controlnet, torch_dtype=torch.float16)
 pipe.to("cuda")

 import torch
+controlnet = HunyuanDiT2DControlNetModel.from_pretrained("Tencent-Hunyuan/HunyuanDiT-v1.2-ControlNet-Diffusers-Pose", torch_dtype=torch.float16)
 pipe = HunyuanDiTControlNetPipeline.from_pretrained("Tencent-Hunyuan/HunyuanDiT-v1.2-Diffusers-Distilled", controlnet=controlnet, torch_dtype=torch.float16)
 pipe.to("cuda")