diff --git a/src/stores/useModelStore.ts b/src/stores/useModelStore.ts index 1505e5b..b3ce352 100644 --- a/src/stores/useModelStore.ts +++ b/src/stores/useModelStore.ts @@ -339,17 +339,23 @@ export const useModelStore = create((set, get) => ({ fineTuning: { ...defaultFineTuning }, setFineTuning: (ft) => set((state) => { const newFT = { ...state.fineTuning, ...ft }; - // Also write to the active model's fineTuning - const models = state.activeModelId - ? state.models.map(m => m.id === state.activeModelId ? { ...m, fineTuning: newFT } : m) + const activeId = state.activeModelId; + const models = activeId + ? state.models.map(m => m.id === activeId ? { ...m, fineTuning: newFT } : m) : state.models; + // Persiste placement por fileName para sobreviver à saída/re-entrada do AR. + const active = models.find(m => m.id === activeId); + if (active) savePlacement(active.fileName, newFT); return { fineTuning: newFT, models }; }), resetFineTuning: () => set((state) => { const reset = { ...defaultFineTuning }; - const models = state.activeModelId - ? state.models.map(m => m.id === state.activeModelId ? { ...m, fineTuning: reset } : m) + const activeId = state.activeModelId; + const models = activeId + ? state.models.map(m => m.id === activeId ? { ...m, fineTuning: reset } : m) : state.models; + const active = models.find(m => m.id === activeId); + if (active) savePlacement(active.fileName, reset); return { fineTuning: reset, models }; }),