Switch TROCR to new inference session

Author: CodeWithKyrian

src/Models/Pretrained/TrOCRPretrainedModel.php

@@ -8,7 +8,7 @@
 use Codewithkyrian\Transformers\Models\ModelArchitecture;
 use Codewithkyrian\Transformers\Utils\AutoConfig;
 use Codewithkyrian\Transformers\Utils\GenerationConfig;
-use OnnxRuntime\InferenceSession;
+use Codewithkyrian\Transformers\Utils\InferenceSession;
 
 class TrOCRPretrainedModel extends PretrainedModel
 {
@@ -29,8 +29,8 @@ public function __construct(
         parent::__construct($config, $session, $modelArchitecture);
 
 
-        $this->numEncoderLayers =  $this->numDecoderLayers = $this->config['decoder_layers'];
-        $this->numEncoderHeads =  $this->numDecoderHeads = $this->config['decoder_attention_heads'];
-        $this->encoderDimKv =  $this->decoderDimKv = $this->config['d_model'] / $this->numDecoderHeads;
+        $this->numEncoderLayers = $this->numDecoderLayers = $this->config['decoder_layers'];
+        $this->numEncoderHeads = $this->numDecoderHeads = $this->config['decoder_attention_heads'];
+        $this->encoderDimKv = $this->decoderDimKv = $this->config['d_model'] / $this->numDecoderHeads;
     }
 }