Skip to content

Commit 1332b8d

Browse files
authored
Merge pull request #453 from ionite34/oops-main
fix directml showing when it shouldnt
2 parents 6170c5e + cebdc28 commit 1332b8d

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

StabilityMatrix.Core/Models/Packages/StableDiffusionDirectMl.cs

+3-5
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,4 @@
1-
using System.Diagnostics.CodeAnalysis;
2-
using System.Text.Json.Nodes;
3-
using System.Text.RegularExpressions;
4-
using NLog;
1+
using NLog;
52
using StabilityMatrix.Core.Attributes;
63
using StabilityMatrix.Core.Helper;
74
using StabilityMatrix.Core.Helper.Cache;
@@ -37,7 +34,8 @@ IPrerequisiteHelper prerequisiteHelper
3734

3835
public override SharedFolderMethod RecommendedSharedFolderMethod => SharedFolderMethod.Symlink;
3936

40-
public override TorchVersion GetRecommendedTorchVersion() => TorchVersion.DirectMl;
37+
public override TorchVersion GetRecommendedTorchVersion() =>
38+
HardwareHelper.PreferDirectML() ? TorchVersion.DirectMl : base.GetRecommendedTorchVersion();
4139

4240
public override PackageDifficulty InstallerSortOrder => PackageDifficulty.Recommended;
4341

0 commit comments

Comments
 (0)