Merge pull request #383 from LykosAI/main

2.7.7.1
LykosAI · Jan 4, 2024 · f03f7d3 · f03f7d3
2 parents 0782fb4 + 1daa7d1
commit f03f7d3
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/StabilityMatrix.Core/Models/Packages/StableDiffusionDirectMl.cs b/StabilityMatrix.Core/Models/Packages/StableDiffusionDirectMl.cs
@@ -1,7 +1,4 @@
-using System.Diagnostics.CodeAnalysis;
-using System.Text.Json.Nodes;
-using System.Text.RegularExpressions;
-using NLog;
+using NLog;
 using StabilityMatrix.Core.Attributes;
 using StabilityMatrix.Core.Helper;
 using StabilityMatrix.Core.Helper.Cache;
@@ -37,7 +34,8 @@ IPrerequisiteHelper prerequisiteHelper
 
     public override SharedFolderMethod RecommendedSharedFolderMethod => SharedFolderMethod.Symlink;
 
-    public override TorchVersion GetRecommendedTorchVersion() => TorchVersion.DirectMl;
+    public override TorchVersion GetRecommendedTorchVersion() =>
+        HardwareHelper.PreferDirectML() ? TorchVersion.DirectMl : base.GetRecommendedTorchVersion();
 
     public override PackageDifficulty InstallerSortOrder => PackageDifficulty.Recommended;