diff --git a/MindIE/MultiModal/Juggernaut-XI-Lightning/README.md b/MindIE/MultiModal/Juggernaut-XI-Lightning/README.md index 6e689ed2a5aa838f3bb3542c59e2833597bf77a7..f88ca763399c94b5249e80ded85ee0c71aaa6bb8 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 | ## 精度验证 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 f43e316b5486733a71edac76debfa60c2cda605b..c4f2cfe196e106c461ee7935cf105d1ed049c123 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 6353806d3f71e2b739caf8df040a9e852530ffc2..4689e7a717904d7c335cfaf2de957bbd7c356b17 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