From 1504a13aa8fb554679608bcf06926ab93ef885b9 Mon Sep 17 00:00:00 2001
From: m30045614 <mazhixin4@huawei.com>
Date: Wed, 17 Jul 2024 15:15:31 +0800
Subject: [PATCH 1/2] =?UTF-8?q?=E5=A2=9E=E5=8A=A0SDXL=E6=94=AF=E6=8C=81?=
 =?UTF-8?q?=E7=9A=84=E6=A1=A3=E4=BD=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../built-in/foundation/stable_diffusion_xl/export_ts.py      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/MindIE/MindIE-Torch/built-in/foundation/stable_diffusion_xl/export_ts.py b/MindIE/MindIE-Torch/built-in/foundation/stable_diffusion_xl/export_ts.py
index e531e8c89a..8336398f9a 100644
--- a/MindIE/MindIE-Torch/built-in/foundation/stable_diffusion_xl/export_ts.py
+++ b/MindIE/MindIE-Torch/built-in/foundation/stable_diffusion_xl/export_ts.py
@@ -715,8 +715,8 @@ if __name__ == "__main__":
     min_height, max_height = 512 // 8, 1024 // 8
     min_width, max_width = 512 // 8, 1664 // 8
     # 动态分档支持的分辨率
-    heights = [1024, 512]
-    widths = [1024, 512]
+    heights = [1024, 512, 936, 768, 576]
+    widths = [1024, 512, 1664, 1360, 1024]
     
     args = parse_arguments()
     if args.soc == "Duo":
-- 
Gitee


From cb2d2339bc277c5a2b42a73c92d8353cc634a1b2 Mon Sep 17 00:00:00 2001
From: m30045614 <mazhixin4@huawei.com>
Date: Wed, 24 Jul 2024 15:14:11 +0800
Subject: [PATCH 2/2] =?UTF-8?q?readme=E4=B8=AD=E5=88=A0=E9=99=A4warmup?=
 =?UTF-8?q?=E8=AF=B4=E6=98=8E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 MindIE/MindIE-Torch/built-in/foundation/DiT/README.md | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/MindIE/MindIE-Torch/built-in/foundation/DiT/README.md b/MindIE/MindIE-Torch/built-in/foundation/DiT/README.md
index 9323b25549..4a14cf3aa3 100644
--- a/MindIE/MindIE-Torch/built-in/foundation/DiT/README.md
+++ b/MindIE/MindIE-Torch/built-in/foundation/DiT/README.md
@@ -150,8 +150,7 @@ latent_size = image_size // 8
       --device 0 \
       --class_label 0 \
       --output_dir ./models \
-      --parallel \
-      --warmup
+      --parallel
       
       # A2
       python3 sample_npu.py \
@@ -160,8 +159,7 @@ latent_size = image_size // 8
       --ckpt ./DiT-XL-2-512x512.pt \
       --device 0 \
       --class_label 0 \
-      --output_dir ./models \
-      --warmup
+      --output_dir ./models
       ```
 
       参数说明：
@@ -173,7 +171,6 @@ latent_size = image_size // 8
       - --class_label：可在0~999中任意指定一个整数，代表image_net的种类
       - --output_dir：上一步骤指定的pt模型输出目录
       - --parallel：【可选】模型使用并行进行推理
-      - --warmup:【可选】，使用warmup可以使模型的推理时间更准确
 
 4. 精度验证
 
-- 
Gitee