From 764321868c3c369827a58a07297c3e2766c22ba9 Mon Sep 17 00:00:00 2001
From: Christian Bastian <80225746+cdb-boop@users.noreply.github.com>
Date: Mon, 25 Mar 2024 19:00:12 -0400
Subject: [PATCH] Move sampling type in Model Info.

---
 __init__.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 64b29fc..b1e128b 100644
--- a/__init__.py
+++ b/__init__.py
@@ -632,12 +632,11 @@ async def get_model_info(request):
         info["Architecture"] = metadata.get("modelspec.architecture", "")
         info["Network Dimension"] = metadata.get("ss_network_dim", "") # features trained
         info["Network Alpha"] = metadata.get("ss_network_alpha", "") # trained features applied
-
+        info["Model Sampling Type"] = metadata.get("modelspec.prediction_type", "")
         clip_skip = metadata.get("ss_clip_skip", "")
         if clip_skip == "None" or clip_skip == "1": # assume 1 means no clip skip
             clip_skip = ""
         info["Clip Skip"] = clip_skip
-        info["Model Sampling Type"] = metadata.get("modelspec.prediction_type", "")
 
         # it is unclear what these are
         #info["Hash SHA256"] = metadata.get("modelspec.hash_sha256", "")