From a5e62d1bfddefad584b6e12f017d56d67ae007c2 Mon Sep 17 00:00:00 2001
From: Robin Huang <robin.j.huang@gmail.com>
Date: Thu, 3 Oct 2024 12:54:28 +0800
Subject: [PATCH] Add model download URL and change to use safetensors.

---
 public/templates/default.json     |  9 +++++++--
 public/templates/image2image.json |  9 +++++++--
 public/templates/upscale.json     | 11 +++++++++--
 3 files changed, 23 insertions(+), 6 deletions(-)

diff --git a/public/templates/default.json b/public/templates/default.json
index a950918e94..657ac107cc 100644
--- a/public/templates/default.json
+++ b/public/templates/default.json
@@ -266,7 +266,7 @@
       ],
       "properties": {},
       "widgets_values": [
-        "v1-5-pruned-emaonly.ckpt"
+        "v1-5-pruned-emaonly.safetensors"
       ]
     }
   ],
@@ -347,5 +347,10 @@
   "groups": [],
   "config": {},
   "extra": {},
-  "version": 0.4
+  "version": 0.4,
+  "models": [{
+    "name": "v1-5-pruned-emaonly.safetensors",
+    "url": "https://huggingface.co/Comfy-Org/stable-diffusion-v1-5-archive/resolve/main/v1-5-pruned-emaonly.safetensors?download=true",
+    "directory": "checkpoints"
+  }]
 }
diff --git a/public/templates/image2image.json b/public/templates/image2image.json
index 8baa1a1a82..81da539d0f 100644
--- a/public/templates/image2image.json
+++ b/public/templates/image2image.json
@@ -330,7 +330,7 @@
         "Node name for S&R": "CheckpointLoaderSimple"
       },
       "widgets_values": [
-        "v1-5-pruned-emaonly.ckpt"
+        "v1-5-pruned-emaonly.safetensors"
       ]
     }
   ],
@@ -438,5 +438,10 @@
   ],
   "config": {},
   "extra": {},
-  "version": 0.4
+  "version": 0.4,
+  "models": [{
+    "name": "v1-5-pruned-emaonly.safetensors",
+    "url": "https://huggingface.co/Comfy-Org/stable-diffusion-v1-5-archive/resolve/main/v1-5-pruned-emaonly.safetensors?download=true",
+    "directory": "checkpoints"
+  }]
 }
diff --git a/public/templates/upscale.json b/public/templates/upscale.json
index 65711beacb..5d568b1c50 100644
--- a/public/templates/upscale.json
+++ b/public/templates/upscale.json
@@ -429,7 +429,7 @@
           "Node name for S&R": "CheckpointLoaderSimple"
         },
         "widgets_values": [
-          "v2-1_768-ema-pruned.ckpt"
+          "v2-1_768-ema-pruned.safetensors"
         ]
       },
       {
@@ -641,5 +641,12 @@
     ],
     "config": {},
     "extra": {},
-    "version": 0.4
+    "version": 0.4,
+    "models": [
+      {
+        "name": "v2-1_768-ema-pruned.safetensors",
+        "url": "https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.safetensors?download=true",
+        "directory": "checkpoints"
+      }
+    ]
   }