fix hires steps 0 and docstring directory images

R3gm · Nov 20, 2023 · 40ac96e · 40ac96e
1 parent 7249d6a
commit 40ac96e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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: