From f2d33327ee5e6dfde0579f1f54e8f6aac846a4c0 Mon Sep 17 00:00:00 2001
From: Joey Ballentine <34788790+joeyballentine@users.noreply.github.com>
Date: Sun, 5 May 2024 16:28:49 -0400
Subject: [PATCH] Fix invoke-spandrel file (#406)

---
 invoke-spandrel.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/invoke-spandrel.py b/invoke-spandrel.py
index 8c8438ab..49b036e3 100644
--- a/invoke-spandrel.py
+++ b/invoke-spandrel.py
@@ -27,7 +27,7 @@ def print_metadata(file: str):
 
         return_success(
             {
-                "architecture": model.architecture,
+                "architecture": model.architecture.name,
                 "tags": model.tags,
                 "scale": model.scale,
                 "inputChannels": model.input_channels,
@@ -45,6 +45,8 @@ def print_metadata(file: str):
             current_python = sys.executable
             os.system(current_python + " -m pip install spandrel")
         else:
-            return_error("PyTorch is not installed. Install PyTorch on your system's Python installation to automatically detect model metadata.")
+            return_error(
+                "PyTorch is not installed. Install PyTorch on your system's Python installation to automatically detect model metadata."
+            )
 
     print_metadata(sys.argv[1])