From 05061a70b383fb00d29dfb27cb36fd7489c110af Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lincoln.stein@gmail.com>
Date: Sun, 28 Aug 2022 23:13:23 -0400
Subject: [PATCH] report errors on non-cuda systems rather than failing
 silently

---
 ldm/simplet2i.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py
index 3b869d4614..ab4778dc44 100644
--- a/ldm/simplet2i.py
+++ b/ldm/simplet2i.py
@@ -20,6 +20,7 @@ from contextlib import contextmanager, nullcontext
 import transformers
 import time
 import re
+import sys
 
 from ldm.util import instantiate_from_config
 from ldm.models.diffusion.ddim import DDIMSampler
@@ -540,6 +541,9 @@ class T2I:
                 # model.to doesn't change the cond_stage_model.device used to move the tokenizer output, so set it here
                 self.model.cond_stage_model.device = self.device
             except AttributeError:
+                import traceback
+                print('Error loading model. Only the CUDA backend is supported',file=sys.stderr)
+                print(traceback.format_exc(),file=sys.stderr)
                 raise SystemExit
 
             self._set_sampler()