Update app.py
Browse files
app.py
CHANGED
@@ -1300,7 +1300,9 @@ with tabs[8]:
|
|
1300 |
st.session_state['history'].append(f"Diffusion Gen: '{prompt_imggen_diff[:30]}...' -> {output_imggen_file_diff}")
|
1301 |
except ImportError: st.error("Diffusers or Torch library not found.")
|
1302 |
except Exception as e: st.error(f"Image generation failed: {e}"); logger.error(f"Diffusion generation failed for {selected_diffusion_model_path}: {e}", exc_info=True)
|
1303 |
-
finally:
|
|
|
|
|
1304 |
|
1305 |
# --- Tab 10: Character Editor ---
|
1306 |
with tabs[9]:
|
|
|
1300 |
st.session_state['history'].append(f"Diffusion Gen: '{prompt_imggen_diff[:30]}...' -> {output_imggen_file_diff}")
|
1301 |
except ImportError: st.error("Diffusers or Torch library not found.")
|
1302 |
except Exception as e: st.error(f"Image generation failed: {e}"); logger.error(f"Diffusion generation failed for {selected_diffusion_model_path}: {e}", exc_info=True)
|
1303 |
+
finally:
|
1304 |
+
if 'pipe' in locals():
|
1305 |
+
del pipe; torch.cuda.empty_cache() if device == "cuda" else None # Clear VRAM
|
1306 |
|
1307 |
# --- Tab 10: Character Editor ---
|
1308 |
with tabs[9]:
|