Merge pull request #1 from ModeratePrawn/main

cpu only img to img from main repo
3 years ago · dd9f9be146
parent a9d28bb756 a32c45353e
commit dd9f9be146
1 changed files with 2 additions and 1 deletions
--- 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