diff --git a/studio/frontend/src/features/images/train/diffusion-train-panel.tsx b/studio/frontend/src/features/images/train/diffusion-train-panel.tsx index 8c4ce6b7c0..45df1d50a6 100644 --- a/studio/frontend/src/features/images/train/diffusion-train-panel.tsx +++ b/studio/frontend/src/features/images/train/diffusion-train-panel.tsx @@ -288,6 +288,16 @@ export function DiffusionTrainPanel({ } }, [family, loadedBaseRepo]); + // The base actually used everywhere (request, deploy, select value). baseChoice can + // briefly hold another family's repo between a family switch and the reseed effect + // (or if that effect is skipped); a raw setBaseChoice(e.target.value)} className={selectClass} aria-label="Base model" @@ -545,7 +555,7 @@ export function DiffusionTrainPanel({ ))} - {baseChoice === CUSTOM_BASE && ( + {effectiveBase === CUSTOM_BASE && ( - {/* Adapter name + trigger */} -
- - setOutputDir(e.target.value)} - className="h-8 text-xs" - /> -
+ {/* Trigger + adapter name (trigger first: it describes the dataset, the name + just labels the output) */} {fullyCaptioned ? (

All {selectedDataset?.image_count} images have captions - no trigger prompt needed. @@ -702,6 +703,16 @@ export function DiffusionTrainPanel({ /> )} +

+ + setOutputDir(e.target.value)} + className="h-8 text-xs" + /> +
{/* Collapsed training settings */}