From 40ac96e29175878fa62614508a6ffe63e96110ad Mon Sep 17 00:00:00 2001
From: Roger Condori <114810545+R3gm@users.noreply.github.com>
Date: Mon, 20 Nov 2023 16:23:16 -0400
Subject: [PATCH] fix hires steps 0 and docstring directory images

---
 stablepy/diffusers_vanilla/model.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/stablepy/diffusers_vanilla/model.py b/stablepy/diffusers_vanilla/model.py
index 9092538..2cd1253 100644
--- a/stablepy/diffusers_vanilla/model.py
+++ b/stablepy/diffusers_vanilla/model.py
@@ -1586,6 +1586,8 @@ def __call__(
                 If you use a notebook, you will be able to display the images generated with this parameter.
             save_generated_images (bool, optional, defaults to True):
                 By default, the generated images are saved in the current location within the 'images' folder. You can disable this with this parameter.
+            image_storage_location (str , optional, defaults to "./images"):
+                The directory where the generated images are saved.
             generator_in_cpu (bool, optional, defaults to False):
                 The generator by default is specified on the GPU. To obtain more consistent results across various environments,
                 it is preferable to use the generator on the CPU.
@@ -2376,6 +2378,9 @@ def __call__(
             hires_pipe.to(self.device)
             torch.cuda.empty_cache()
             gc.collect()
+        else:
+            hires_params_config = {}
+            hires_pipe = None
 
         # Debug info
         try: