From 1b60dae64e6bc0bc00d56215006bfd1a78f3840b Mon Sep 17 00:00:00 2001 From: zhoufan2956 Date: Sun, 27 Apr 2025 17:15:40 +0800 Subject: [PATCH 1/2] update_cache --- .../juggernaut_xi_lightning/unet/transformer_blocks.py | 6 +++--- MindIE/MultiModal/Stable-Audio-Open-1.0/requirements.txt | 3 +-- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/MindIE/MultiModal/Juggernaut-XI-Lightning/juggernaut_xi_lightning/unet/transformer_blocks.py b/MindIE/MultiModal/Juggernaut-XI-Lightning/juggernaut_xi_lightning/unet/transformer_blocks.py index f43e316b54..c4f2cfe196 100644 --- a/MindIE/MultiModal/Juggernaut-XI-Lightning/juggernaut_xi_lightning/unet/transformer_blocks.py +++ b/MindIE/MultiModal/Juggernaut-XI-Lightning/juggernaut_xi_lightning/unet/transformer_blocks.py @@ -229,9 +229,9 @@ class BasicTransformerBlock(nn.Module): self.cross_alpha_max = -float('inf') self.last_cross = None - self.bound = [10, 2] - self.forcefresh = 6 - self.totalstep = 50 + self.bound = [6, 0] + self.forcefresh = 4 + self.totalstep = 12 def set_chunk_feed_forward(self, chunk_size: Optional[int], dim: int = 0): # Sets chunk feed-forward diff --git a/MindIE/MultiModal/Stable-Audio-Open-1.0/requirements.txt b/MindIE/MultiModal/Stable-Audio-Open-1.0/requirements.txt index 6353806d3f..4689e7a717 100644 --- a/MindIE/MultiModal/Stable-Audio-Open-1.0/requirements.txt +++ b/MindIE/MultiModal/Stable-Audio-Open-1.0/requirements.txt @@ -2,5 +2,4 @@ torch==2.1.0 torchsde==0.2.6 diffusers==0.30.0 transformers==4.40.0 -soundfile==0.12.1 -torch_npu==2.1.0.post6 \ No newline at end of file +soundfile==0.12.1 \ No newline at end of file -- Gitee From 5ec8ec1f3f0fe2a12c56a48e80f70df4a6b89060 Mon Sep 17 00:00:00 2001 From: zhoufan2956 Date: Sun, 27 Apr 2025 17:21:22 +0800 Subject: [PATCH 2/2] update_cache --- MindIE/MultiModal/Juggernaut-XI-Lightning/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/MindIE/MultiModal/Juggernaut-XI-Lightning/README.md b/MindIE/MultiModal/Juggernaut-XI-Lightning/README.md index 6e689ed2a5..f88ca76339 100644 --- a/MindIE/MultiModal/Juggernaut-XI-Lightning/README.md +++ b/MindIE/MultiModal/Juggernaut-XI-Lightning/README.md @@ -168,7 +168,7 @@ https://huggingface.co/Rundiffusion/Juggernaut-XI-Lightning | 800I A2 32G | 分辨率 | 迭代次数 | 单卡推理 | 双卡推理| |-------|------|------|--------|-----------| |无损优化| 1024x1024 | 12 | 2.09s | 1.99s | - |算法优化| 1024x1024 | 12 | 1.67s | 1.58s | + |算法优化| 1024x1024 | 12 | 1.67s | 1.44s | ## 精度验证 -- Gitee