From d3f1ebcbd71ec5521142c55f617e92303ffd872d Mon Sep 17 00:00:00 2001
From: SpandexWizard <80999980+SpandexWizard@users.noreply.github.com>
Date: Thu, 1 Sep 2022 03:52:40 -0500
Subject: [PATCH] added cpu support to img2img.py

---
 scripts/img2img.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/scripts/img2img.py b/scripts/img2img.py
index 421e215..285fef7 100644
--- a/scripts/img2img.py
+++ b/scripts/img2img.py
@@ -40,7 +40,8 @@ def load_model_from_config(config, ckpt, verbose=False):
         print("unexpected keys:")
         print(u)
 
-    model.cuda()
+    if torch.cuda.is_available():
+        model.cuda()
     model.eval()
     return model