Skip to content

Commit d07b182

Browse files
Fix autoTokenizer sending stale arguments
1 parent c91a888 commit d07b182

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/Pipelines/Task.php

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77
use Codewithkyrian\Transformers\Models\Auto\AutoModel;
88
use Codewithkyrian\Transformers\Models\Auto\AutoModelForCausalLM;
99
use Codewithkyrian\Transformers\Models\Auto\AutoModelForImageClassification;
10-
use Codewithkyrian\Transformers\Models\Auto\AutoModelForImageToImage;
1110
use Codewithkyrian\Transformers\Models\Auto\AutoModelForImageFeatureExtraction;
11+
use Codewithkyrian\Transformers\Models\Auto\AutoModelForImageToImage;
1212
use Codewithkyrian\Transformers\Models\Auto\AutoModelForMaskedLM;
1313
use Codewithkyrian\Transformers\Models\Auto\AutoModelForObjectDetection;
1414
use Codewithkyrian\Transformers\Models\Auto\AutoModelForQuestionAnswering;
@@ -212,7 +212,7 @@ public function autoTokenizer(
212212
self::Ner,
213213
self::ImageToText,
214214
self::ZeroShotImageClassification,
215-
self::ZeroShotObjectDetection => AutoTokenizer::fromPretrained($modelNameOrPath, $quantized, $config, $cacheDir, $revision, null, $output),
215+
self::ZeroShotObjectDetection => AutoTokenizer::fromPretrained($modelNameOrPath, $cacheDir, $revision, null, $output),
216216
};
217217
}
218218

0 commit comments

Comments
 (0)