Upload 2 files

app.py

@@ -132,8 +132,8 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scal
     global controlnet
     global controlnet_union
     try:
-        good_vae.to("cuda")
-        taef1.to("cuda")
+        #good_vae.to("cuda")
+        #taef1.to("cuda")
         pipe.to("cuda")
         generator = torch.Generator(device="cuda").manual_seed(seed)
         
@@ -338,6 +338,10 @@ css = '''
 .card_internal{display: flex;height: 100px;margin-top: .5em}
 .card_internal img{margin-right: 1em}
 .styler{--form-gap-width: 0px !important}
+#progress{height:30px}
+#progress .generating{display:none}
+.progress-container {width: 100%;height: 30px;background-color: #f0f0f0;border-radius: 15px;overflow: hidden;margin-bottom: 20px}
+.progress-bar {height: 100%;background-color: #4f46e5;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out}
 #model-info {text-align: center; !important}
 '''
 with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:

mod.py

@@ -75,6 +75,8 @@ def is_repo_exists(repo_id):
 
 def clear_cache():
     torch.cuda.empty_cache()
+    torch.cuda.reset_max_memory_allocated()
+    torch.cuda.reset_peak_memory_stats()
     gc.collect()