Deploy from GitHub main
Browse files
src/ui/compi_phase3_final_dashboard.py
CHANGED
|
@@ -16,6 +16,7 @@ Features:
|
|
| 16 |
- True fusion engine: real processing for all inputs, intelligent generation mode selection
|
| 17 |
"""
|
| 18 |
|
|
|
|
| 19 |
import os
|
| 20 |
import io
|
| 21 |
import csv
|
|
@@ -1290,6 +1291,14 @@ with tab_inputs:
|
|
| 1290 |
paths.append(path)
|
| 1291 |
images.append(img)
|
| 1292 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1293 |
# Log generation
|
| 1294 |
rows.append({
|
| 1295 |
"filepath": path,
|
|
@@ -1334,6 +1343,17 @@ with tab_inputs:
|
|
| 1334 |
log_rows(rows, RUNLOG)
|
| 1335 |
st.success(f"Saved {len(rows)} image(s). Run log updated: {RUNLOG}")
|
| 1336 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1337 |
# ==================== GALLERY & ANNOTATE TAB (Phase 3.D) ====================
|
| 1338 |
|
| 1339 |
with tab_gallery:
|
|
|
|
| 16 |
- True fusion engine: real processing for all inputs, intelligent generation mode selection
|
| 17 |
"""
|
| 18 |
|
| 19 |
+
import gc
|
| 20 |
import os
|
| 21 |
import io
|
| 22 |
import csv
|
|
|
|
| 1291 |
paths.append(path)
|
| 1292 |
images.append(img)
|
| 1293 |
|
| 1294 |
+
# EXPLICIT VRAM CLEANUP after successful generation
|
| 1295 |
+
if DEVICE == "cuda":
|
| 1296 |
+
torch.cuda.empty_cache()
|
| 1297 |
+
torch.cuda.synchronize()
|
| 1298 |
+
# Force garbage collection to help with Python object cleanup
|
| 1299 |
+
import gc
|
| 1300 |
+
gc.collect()
|
| 1301 |
+
|
| 1302 |
# Log generation
|
| 1303 |
rows.append({
|
| 1304 |
"filepath": path,
|
|
|
|
| 1343 |
log_rows(rows, RUNLOG)
|
| 1344 |
st.success(f"Saved {len(rows)} image(s). Run log updated: {RUNLOG}")
|
| 1345 |
|
| 1346 |
+
# FINAL VRAM CLEANUP after entire batch
|
| 1347 |
+
if DEVICE == "cuda":
|
| 1348 |
+
torch.cuda.empty_cache()
|
| 1349 |
+
torch.cuda.synchronize()
|
| 1350 |
+
import gc
|
| 1351 |
+
gc.collect()
|
| 1352 |
+
# Show updated VRAM usage
|
| 1353 |
+
used_after = vram_used_gb()
|
| 1354 |
+
if used_after:
|
| 1355 |
+
st.info(f"🧹 VRAM cleaned • Current usage: {used_after:.2f} GB")
|
| 1356 |
+
|
| 1357 |
# ==================== GALLERY & ANNOTATE TAB (Phase 3.D) ====================
|
| 1358 |
|
| 1359 |
with tab_gallery:
|