refine

microsoft · Nov 29, 2023 · a32b2e1 · a32b2e1
1 parent 1f4ee3e
commit a32b2e1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_utils.py b/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_utils.py
@@ -255,7 +255,7 @@ def parse_arguments(is_xl: bool, parser):
                 print("[I] Use --denoising_steps=4 (no more than 8) for sdxl-turbo.")
                 args.denoising_steps = 4
             if not args.disable_refiner:
-                print("[I] sdxl-turbo cannot use with SDXL refiner.")
+                print("[I] Disable SDXL refiner to run sdxl-turbo.")
                 args.disable_refiner = True
 
         if args.lcm and args.scheduler != "LCM":

diff --git a/onnxruntime/python/tools/transformers/models/stable_diffusion/diffusion_models.py b/onnxruntime/python/tools/transformers/models/stable_diffusion/diffusion_models.py
@@ -223,7 +223,7 @@ def unet_embedding_dim(self):
             return 1024
         elif self.is_xl_base_or_turbo():
             return 2048
-        elif self.version == "xl-1.0" and self.is_xl_refiner():
+        elif self.is_xl_refiner():
             return 1280
         else:
             raise ValueError(f"Invalid version {self.version}")