diff --git a/app/ui/features/training/index.js b/app/ui/features/training/index.js index 98af78b9..ab9bbcce 100644 --- a/app/ui/features/training/index.js +++ b/app/ui/features/training/index.js @@ -33,7 +33,7 @@ export default { advanced: false, runs: [], chosenRun: "", record: null, run: { shown: false, active: false, state: "idle", loss: [], smooth: [], step: 0, total: 0, elapsed: 0, message: "", error: "", lora: "", before: "", runId: "", paused: null }, previews: [], holdPause: false, awaitStart: false, peek: (item) => peek(item), - pause: () => pause(), resume: () => launch(ui.run.runId), restore: (id) => { ui.run.loss = []; return launch(id); }, previewNow: () => previewNow(), + pause: () => pause(), resume: () => launch(ui.run.runId), restore: (id) => { ui.run.loss = []; ui.run.runId = id; return launch(id); }, previewNow: () => previewNow(), monitorBox: null, redraw: () => draw(), pollTaggers: () => pollTaggers(true), loadRecord: (id) => loadRecord(id), loadRuns: () => loadRuns(), start: () => start(), stop: () => stop(), /** Ask for the frame counts of clips not known yet; `then` runs when new numbers arrived. */ diff --git a/app/ui/features/training/results.js b/app/ui/features/training/results.js index 0eb832c2..ccb81aac 100644 --- a/app/ui/features/training/results.js +++ b/app/ui/features/training/results.js @@ -5,7 +5,7 @@ import { clock, fmt, monitor } from "./train.js"; const STATES = { done: "Finished", error: "Failed", stopped: "Stopped", training: "Running" }; export function resultsStep({ app, ui }) { - const runs = ui.runs; + const runs = ui.runs.filter((r) => !(ui.run.active && r.run === ui.run.runId)); // the run going on is shown above, not a second time from its old record const live = c.region.stack({ gap: "md", children: [monitor(ui)].filter(Boolean) }); // the run going on now, redrawn on its own ui.monitorBox = live; if (!runs.length) return ui.run.shown ? live : c.emptyState.default({ title: "No runs yet", hint: "A run you start in the Train step is kept here, with its loss graph." });