File tree 1 file changed +3
-5
lines changed
StabilityMatrix.Core/Models/Packages
1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change 1
- using System . Diagnostics . CodeAnalysis ;
2
- using System . Text . Json . Nodes ;
3
- using System . Text . RegularExpressions ;
4
- using NLog ;
1
+ using NLog ;
5
2
using StabilityMatrix . Core . Attributes ;
6
3
using StabilityMatrix . Core . Helper ;
7
4
using StabilityMatrix . Core . Helper . Cache ;
@@ -37,7 +34,8 @@ IPrerequisiteHelper prerequisiteHelper
37
34
38
35
public override SharedFolderMethod RecommendedSharedFolderMethod => SharedFolderMethod . Symlink ;
39
36
40
- public override TorchVersion GetRecommendedTorchVersion ( ) => TorchVersion . DirectMl ;
37
+ public override TorchVersion GetRecommendedTorchVersion ( ) =>
38
+ HardwareHelper . PreferDirectML ( ) ? TorchVersion . DirectMl : base . GetRecommendedTorchVersion ( ) ;
41
39
42
40
public override PackageDifficulty InstallerSortOrder => PackageDifficulty . Recommended ;
43
41
You can’t perform that action at this time.
0 commit comments