awacke1 commited on
Commit
899f03b
·
verified ·
1 Parent(s): a7732cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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: if 'pipe' in locals(): del pipe; torch.cuda.empty_cache() if device == "cuda" else None # Clear VRAM
 
 
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]: