Skip to content

Commit 182c415

Browse files
committed
Add dispose for image folder and prompt vms
1 parent 5f7fb4c commit 182c415

File tree

2 files changed

+12
-9
lines changed

2 files changed

+12
-9
lines changed

StabilityMatrix.Avalonia/ViewModels/Base/InferenceGenerationViewModelBase.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ RunningPackageService runningPackageService
9393
ClientManager = inferenceClientManager;
9494

9595
ImageGalleryCardViewModel = vmFactory.Get<ImageGalleryCardViewModel>();
96-
ImageFolderCardViewModel = vmFactory.Get<ImageFolderCardViewModel>();
96+
ImageFolderCardViewModel = AddDisposable(vmFactory.Get<ImageFolderCardViewModel>());
9797

9898
GenerateImageCommand.WithConditionalNotificationErrorHandler(notificationService);
9999
}

StabilityMatrix.Avalonia/ViewModels/Inference/InferenceTextToImageViewModel.cs

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,8 @@ RunningPackageService runningPackageService
8383
samplerCard.DenoiseStrength = 1.0d;
8484
});
8585

86-
PromptCardViewModel = vmFactory.Get<PromptCardViewModel>();
86+
PromptCardViewModel = AddDisposable(vmFactory.Get<PromptCardViewModel>());
87+
8788
BatchSizeCardViewModel = vmFactory.Get<BatchSizeCardViewModel>();
8889

8990
ModulesCardViewModel = vmFactory.Get<StackEditableCardViewModel>(modulesCard =>
@@ -109,13 +110,15 @@ RunningPackageService runningPackageService
109110
);
110111

111112
// When refiner is provided in model card, enable for sampler
112-
ModelCardViewModel
113-
.WhenPropertyChanged(x => x.IsRefinerSelectionEnabled)
114-
.Subscribe(e =>
115-
{
116-
SamplerCardViewModel.IsRefinerStepsEnabled =
117-
e.Sender is { IsRefinerSelectionEnabled: true, SelectedRefiner: not null };
118-
});
113+
AddDisposable(
114+
ModelCardViewModel
115+
.WhenPropertyChanged(x => x.IsRefinerSelectionEnabled)
116+
.Subscribe(e =>
117+
{
118+
SamplerCardViewModel.IsRefinerStepsEnabled =
119+
e.Sender is { IsRefinerSelectionEnabled: true, SelectedRefiner: not null };
120+
})
121+
);
119122
}
120123

121124
/// <inheritdoc />

0 commit comments

Comments
 (0)