adding torch empty cache

app.py

@@ -227,7 +227,7 @@ def re_run_inference(config, config_history, pipeline, progress=gr.Progress(trac
         return "Please select a model AND a scheduler.", "Please select a model AND a scheduler.", None, pipeline
     
     if pipeline == None:
-        return "Please run full inference first.", "Please select a model AND a scheduler.", None, pipeline
+        return "Please run full inference first.", "Please run full inference first.", None, pipeline
 
     # # MANUAL SEED/GENERATOR - we probably don't need that again?
     # if config["manual_seed"] is None or config["manual_seed"] == '' or int(config["manual_seed"]) < 0:
@@ -269,14 +269,15 @@ def re_run_inference(config, config_history, pipeline, progress=gr.Progress(trac
 
     # expected output: out_image, out_config_history, config_history, pipeline
     return image[0], dict_list_to_markdown_table(config_history), config_history, pipeline
-    
-    
+        
 def run_inference(config, config_history, pipeline, progress=gr.Progress(track_tqdm=True)):
     
     if str(config["model"]) != 'None' and str(config["model"]) != 'null' and str(config["model"]) != '' and str(config["scheduler"]) != 'None':
 
         progress(1, desc="Initializing pipeline...")
         
+        torch.cuda.empty_cache()
+        
         torch.backends.cuda.matmul.allow_tf32 = get_bool(config["allow_tensorfloat32"]) # Use TensorFloat-32 as of https://huggingface.co/docs/diffusers/main/en/optimization/fp16 faster, but slightly less accurate computations
                 
         # INIT PIPELINE
@@ -427,7 +428,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
     pipeline = gr.State()
 
     gr.Markdown('''## Text-2-Image Playground
-                <small>by <a target="_blank" href="https://www.linkedin.com/in/nickyreinert/">Nicky Reinert</a> | 
+                <small>by <a target="_blank" href="https://nickyreinert.de/">Nicky Reinert</a> | 
                 home base: https://huggingface.co/spaces/n42/pictero
                 </small>''')
     gr.Markdown("### Device specific settings")