diff --git a/hook_HYDiT_utils.py b/hook_HYDiT_utils.py index 84a70c5..a37f9d4 100644 --- a/hook_HYDiT_utils.py +++ b/hook_HYDiT_utils.py @@ -73,9 +73,9 @@ def easy_sample_images( prompt="A photo of a girl with a hat on a sunny day", negative_prompt="", batch_size=1, - guidance_scale=5.0, + guidance_scale=2.0, infer_steps=20, - sampler='ddpm', + sampler='dpmpp_2m_karras', train_steps=0, ): from hydit.diffusion.pipeline import StableDiffusionPipeline diff --git a/mz_train_tools_core_HYDiT.py b/mz_train_tools_core_HYDiT.py index d1c4623..3680387 100644 --- a/mz_train_tools_core_HYDiT.py +++ b/mz_train_tools_core_HYDiT.py @@ -169,10 +169,10 @@ def MZ_HYDiTDatasetConfig_call(args={}): HYDiT_MODEL = { "HunyuanDiT/t2i/model/pytorch_model_ema.pt": { - "url": "https://www.modelscope.cn/api/v1/models/modelscope/HunyuanDiT/repo?Revision=master&FilePath=t2i%2Fmodel%2Fpytorch_model_ema.pt", + "url": "https://www.modelscope.cn/models/wailovet/hy1.1/resolve/master/t2i/model/pytorch_model_module.pt", }, "HunyuanDiT/t2i/model/pytorch_model_module.pt": { - "url": "https://www.modelscope.cn/api/v1/models/modelscope/HunyuanDiT/repo?Revision=master&FilePath=t2i%2Fmodel%2Fpytorch_model_module.pt", + "url": "https://www.modelscope.cn/models/wailovet/hy1.1/resolve/master/t2i/model/pytorch_model_module.pt", }, "HunyuanDiT/t2i/sdxl-vae-fp16-fix/diffusion_pytorch_model.safetensors": { "url": "https://www.modelscope.cn/api/v1/models/modelscope/HunyuanDiT/repo?Revision=master&FilePath=t2i%2Fsdxl-vae-fp16-fix%2Fdiffusion_pytorch_model.safetensors",