From 5d5f22e6e1ba198c4b32c7d794c8ba6d77d39d23 Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sat, 12 Aug 2023 08:07:13 +0000
Subject: [PATCH] remove invalid downloaded files

---
 modules/modelloader.py | 3 +++
 webui.py               | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/modules/modelloader.py b/modules/modelloader.py
index 56a264937..6527e9efd 100644
--- a/modules/modelloader.py
+++ b/modules/modelloader.py
@@ -35,6 +35,9 @@ def download_civit_model(model_url: str, model_name: str, model_path: str, previ
                     written = written + len(data)
                     f.write(data)
                     progress.update(task, advance=block_size, description="Downloading")
+        if written < 1024 * 1024 * 1024: # min threshold
+            os.remove(model_file)
+            raise ValueError(f'removed invalid download: bytes={written}')
         if preview is not None:
             preview_file = os.path.splitext(model_file)[0] + '.jpg'
             preview.save(preview_file)
diff --git a/webui.py b/webui.py
index 193b544c8..57a37c9a2 100644
--- a/webui.py
+++ b/webui.py
@@ -18,7 +18,8 @@ local_url = None
 errors.log.debug('Loading Torch')
 import torch # pylint: disable=C0411
 try:
-    errors.log.debug(f'Torch init: {torch.sum(torch.randn(2, 2)).to(0) > 0}') # fix silly pytorch_lightning issue
+    rnd = torch.sum(torch.randn(2, 2)).to(0)
+    errors.log.debug(f'Torch init: {rnd / rnd == 1.0}') # fix silly pytorch_lightning issue
 except Exception:
     pass
 try: