From a763e2cd1d407e127c93011bfb7f19c3f326e7dc Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Sat, 30 Dec 2023 22:53:01 -0800
Subject: [PATCH] Dont show onetrainer unless nvidia

---
 StabilityMatrix.Core/Models/Packages/OneTrainer.cs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/StabilityMatrix.Core/Models/Packages/OneTrainer.cs b/StabilityMatrix.Core/Models/Packages/OneTrainer.cs
index fd339fa4..6ec84252 100644
--- a/StabilityMatrix.Core/Models/Packages/OneTrainer.cs
+++ b/StabilityMatrix.Core/Models/Packages/OneTrainer.cs
@@ -3,6 +3,7 @@ using System.Text.RegularExpressions;
 using StabilityMatrix.Core.Attributes;
 using StabilityMatrix.Core.Helper;
 using StabilityMatrix.Core.Helper.Cache;
+using StabilityMatrix.Core.Helper.HardwareInfo;
 using StabilityMatrix.Core.Models.Progress;
 using StabilityMatrix.Core.Processes;
 using StabilityMatrix.Core.Python;
@@ -33,6 +34,8 @@ public class OneTrainer(
     public override string OutputFolderName => string.Empty;
     public override SharedFolderMethod RecommendedSharedFolderMethod => SharedFolderMethod.None;
     public override IEnumerable<TorchVersion> AvailableTorchVersions => [TorchVersion.Cuda];
+    public override bool IsCompatible => HardwareHelper.HasNvidiaGpu();
+
     public override IEnumerable<SharedFolderMethod> AvailableSharedFolderMethods =>
         new[] { SharedFolderMethod.None };
     public override PackageDifficulty InstallerSortOrder => PackageDifficulty.Nightmare;