Fix download command bug

README.md

@@ -164,6 +164,10 @@ Explanation of Arguments:
       models are
       smaller and faster, but may have slightly lower accuracy. Use -q as a shortcut in the command.
 
+> [!DANGER]
+> Remember to add your cache directory to your `.gitignore` file to avoid committing the downloaded models to your git
+> repository.
+
 ## Supported tasks/models
 
 This package is a WIP, but here's a list of tasks and architectures currently tested and supported by Transformers PHP.

src/Commands/DownloadModelCommand.php

@@ -9,6 +9,7 @@
 use Codewithkyrian\Transformers\Models\Auto\AutoModelForCausalLM;
 use Codewithkyrian\Transformers\Models\Auto\AutoModelForSeq2SeqLM;
 use Codewithkyrian\Transformers\Models\Auto\AutoModelForSequenceClassification;
+use Codewithkyrian\Transformers\Transformers;
 use Symfony\Component\Console\Attribute\AsCommand;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputArgument;
@@ -35,8 +36,7 @@ protected function configure(): void
             'cache-dir',
             'c',
             InputOption::VALUE_OPTIONAL,
-            'The directory to cache the model in.',
-            'models'
+            'The directory to cache the model in.'
         );
 
         $this->addOption(
@@ -51,6 +51,8 @@ protected function configure(): void
 
     protected function execute(InputInterface $input, OutputInterface $output): int
     {
+        Transformers::configure();
+
         $output->writeln('✔ Downloading model...');
 
         $model = $input->getArgument('model');
@@ -63,10 +65,10 @@ protected function execute(InputInterface $input, OutputInterface $output): int
         try {
             // TODO: Verify the tasks and corresponding AutoModel classes
             $model = match ($task) {
-                'text-generation' => AutoModelForCausalLM::fromPretrained($model, $quantized, $cacheDir),
-                'text-classification', 'sentiment-analysis' => AutoModelForSequenceClassification::fromPretrained($model, $quantized, $cacheDir),
-                'translation' => AutoModelForSeq2SeqLM::fromPretrained($model, $quantized, $cacheDir),
-                default => AutoModel::fromPretrained($model, $quantized, $cacheDir),
+                'text-generation' => AutoModelForCausalLM::fromPretrained($model, $quantized, cacheDir: $cacheDir),
+                'text-classification', 'sentiment-analysis' => AutoModelForSequenceClassification::fromPretrained($model, $quantized, cacheDir: $cacheDir),
+                'translation' => AutoModelForSeq2SeqLM::fromPretrained($model, $quantized, cacheDir: $cacheDir),
+                default => AutoModel::fromPretrained($model, $quantized, cacheDir: $cacheDir),
             };
 
             $output->writeln('✔ Model downloaded successfully.');

src/Utils/Hub.php

@@ -59,6 +59,7 @@ public static function getFile(
     {
         # Local cache and file paths
         $cacheDir ??= Transformers::$cacheDir;
+
         $filePath = joinPaths($cacheDir, $pathOrRepoID, $subFolder, $fileName);
 
         # Check if file already exists