0gRddlZddlZddlZddlZddlmZdZGddZdS)N)AutoPipelineForText2Imagectj|tj|tj|t |t jd<dS)NPL_GLOBAL_SEED)randomseednptorch manual_seedstrosenviron)rs U/apdcephfs_cq8/share_1367250/huiwenshi/repos/Hunyuan3D-2-spaces/hy3dgen/text2image.pyseed_everythingr"sP KINN4 d#&t99BJ   cReZdZ ddZdZejd dZdS) HunyuanDiTPipeline3Tencent-Hunyuan/HunyuanDiT-v1.1-Diffusers-Distilledcudac||_tj|tjddg||_d|_d|_dS)NTzblocks.(16|17|18|19)) torch_dtype enable_pagpag_applied_layersu#,白色背景,3D风格,最佳质量u~文本,特写,裁剪,出框,最差质量,低质量,JPEG伪影,PGLY,重复,病态,残缺,多余的手指,变异的手,画得不好的手,画得不好的脸,变异,畸形,模糊,脱水,糟糕的解剖学,糟糕的比例,多余的肢体,克隆的脸,毁容,恶心的比例,畸形的肢体,缺失的手臂,缺失的腿,额外的手臂,额外的腿,融合的手指,手指太多,长脖子) devicerfrom_pretrainedr float16topipepos_txtneg_txt)self model_pathrs r__init__zHunyuanDiTPipeline.__init__*s\  -=   67    "V** = ] rc tjdtj|jjd|j_tj|jj}|ddddd d |d d d }dS) NhighT) fullgraphru美少女战士u模糊?Fpromptnegative_promptnum_inference_steps pag_scalewidthheight generator return_dictr)r set_float32_matmul_precisioncompiler transformer Generatorr)r r1out_imgs rr4zHunyuanDiTPipeline.compile<s *6222 % di.Ct T T T O49+;<<< ))$$ "       rrc t|tj|jj}|t |}||j|z|jdddd|ddd}|S)Nr&r(r)Fr*r) rr r6rrr intrr)r r+rr1r7s r__call__zHunyuanDiTPipeline.__call__MsO49+;<<< ))#d))44 ))<& L "       rN)rr)r)__name__ __module__ __qualname__r"r4r no_gradr;rrrr)slI]]]]$"U]___rr) r rnumpyrr diffusersrrrr@rrrCs4  //////---3333333333r