From 15476ba608751d0fe32456c016018c59274ea3e0 Mon Sep 17 00:00:00 2001
From: Tianlei Wu <tlwu@microsoft.com>
Date: Tue, 28 Nov 2023 21:33:35 +0000
Subject: [PATCH] add metadata of device and cuda version

---
 .../tools/transformers/models/stable_diffusion/demo_utils.py    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_utils.py b/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_utils.py
index 82c593f5930eb..f0c83fc507ae4 100644
--- a/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_utils.py
+++ b/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_utils.py
@@ -326,6 +326,8 @@ def get_metadata(args, is_xl: bool = False) -> Dict[str, Any]:
         except PackageNotFoundError:
             continue
     metadata["packages"] = packages
+    metadata["device"] = torch.cuda.get_device_name()
+    metadata["torch.version.cuda"] = torch.version.cuda
 
     return metadata