Move the onnx subfolder for model files to be defaulted on the argu…

…ment
CodeWithKyrian · May 15, 2024 · 115a397 · 115a397
1 parent d83ca9b
commit 115a397
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Models/Pretrained/PretrainedModel.php b/src/Models/Pretrained/PretrainedModel.php
@@ -236,13 +236,13 @@ public static function constructSession(
         string           $fileName,
         ?string          $cacheDir = null,
         string           $revision = 'main',
-        string           $subFolder = '',
+        string           $subFolder = 'onnx',
         bool             $fatal = true,
         ?callable $onProgress = null,
                          ...$sessionOptions
     ): ?InferenceSession
     {
-        $modelFileName = sprintf('onnx/%s.onnx', $fileName);
+        $modelFileName = "$fileName.onnx";
 
         $file = Hub::getFile($modelNameOrPath, $modelFileName, $cacheDir, $revision, $subFolder, $fatal, $onProgress);