From c52c63128b6e1bc270c2669a69115bd2a295d17a Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Thu, 15 Jun 2023 14:08:12 -0400
Subject: [PATCH] handle sending of deleted images

---
 html/locale_en.json                        |  2 +-
 modules/generation_parameters_copypaste.py |  5 +++++
 modules/sd_hijack.py                       |  1 -
 modules/sd_models.py                       | 11 +++++++++--
 modules/shared.py                          |  2 +-
 5 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/html/locale_en.json b/html/locale_en.json
index 7ac6f6096..6fa43cbfe 100644
--- a/html/locale_en.json
+++ b/html/locale_en.json
@@ -308,7 +308,7 @@
   {"id":"","label":"ipex","localized":"","hint":""},
   {"id":"","label":"Model compile verbose mode","localized":"","hint":""},
   {"id":"","label":"Model compile suppress errors","localized":"","hint":""},
-  {"id":"","label":"Disable Torch memory garbage collection (experimental)","localized":"","hint":""},
+  {"id":"","label":"Disable Torch memory garbage collection","localized":"","hint":""},
   {"id":"","label":"Directory for temporary images; leave empty for default","localized":"","hint":""},
   {"id":"","label":"Cleanup non-default temporary directory when starting webui","localized":"","hint":""},
   {"id":"","label":"Path to directory with stable diffusion checkpoints","localized":"","hint":""},
diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index fcb0db36e..2150c3d96 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -59,6 +59,11 @@ def image_from_url_text(filedata):
         is_in_right_dir = ui_tempdir.check_tmp_file(shared.demo, filename)
         if is_in_right_dir:
             filename = filename.rsplit('?', 1)[0]
+            if not os.path.exists(filename):
+                shared.log.error(f'Image file not found: {filename}')
+                image = Image.new('RGB', (512, 512))
+                image.info['parameters'] = f'Image file not found: {filename}'
+                return image
             image = Image.open(filename)
             geninfo, _items = images.read_info_from_image(image)
             image.info['parameters'] = geninfo
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index 93e4492e5..420fbcf18 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -1,4 +1,3 @@
-import sys
 from types import MethodType
 import torch
 from torch.nn.functional import silu
diff --git a/modules/sd_models.py b/modules/sd_models.py
index 7f31a75c5..79a0d9091 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -541,7 +541,6 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None, timer=None)
         sd_hijack.model_hijack.undo_hijack(model_data.sd_model)
         current_checkpoint_info = model_data.sd_model.sd_checkpoint_info
         unload_model_weights()
-        model_data.sd_model = None
     do_inpainting_hijack()
     devices.set_cuda_params()
     if already_loaded_state_dict is not None:
@@ -668,8 +667,16 @@ def unload_model_weights(sd_model=None, _info=None):
         model_data.sd_model.to(devices.cpu)
         if shared.backend == shared.Backend.ORIGINAL:
             sd_hijack.model_hijack.undo_hijack(model_data.sd_model)
-        model_data.sd_model = None
         sd_model = None
+        """
+        if hasattr(model_data.sd_model, 'model'):
+            del model_data.sd_model.model
+        if hasattr(model_data.sd_model, 'first_stage_model'):
+            del model_data.sd_model.first_stage_model
+        if hasattr(model_data.sd_model, 'cond_stage_model'):
+            del model_data.sd_model.cond_stage_model           
+        """
+        model_data.sd_model = None
         devices.torch_gc(force=True)
         shared.log.debug(f'Model weights unloaded: {memory_stats()}')
     return sd_model
diff --git a/modules/shared.py b/modules/shared.py
index 6dd22bd6a..35af1734c 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -331,7 +331,7 @@ options_templates.update(options_section(('cuda', "Compute Settings"), {
     "cuda_compile_mode": OptionInfo("none", "Model compile mode (experimental)", gr.Radio, lambda: {"choices": ['none', 'inductor', 'cudagraphs', 'aot_ts_nvfuser', 'hidet', 'ipex']}),
     "cuda_compile_verbose": OptionInfo(False, "Model compile verbose mode"),
     "cuda_compile_errors": OptionInfo(True, "Model compile suppress errors"),
-    "disable_gc": OptionInfo(False, "Disable Torch memory garbage collection (experimental)"),
+    "disable_gc": OptionInfo(False, "Disable Torch memory garbage collection"),
 }))
 
 options_templates.update(options_section(('system-paths', "System Paths"), {