Fix invoke-spandrel file (#406)

OpenModelDB · May 5, 2024 · f2d3332 · f2d3332
1 parent aff9094
commit f2d3332
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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])