Remove unnecessary check from WavLM (refs #1977)

Author: a2d8a4v

include/ctranslate2/layers/wavlm.h

@@ -80,17 +80,6 @@ namespace ctranslate2 {
         return 1024;
       }
 
-      bool is_encoded(const StorageView& features) const {
-        // Input features shape: [batch_size, input_size, input_time]
-        // Encoder output shape: [batch_size, input_time // 2, output_size]
-        //
-        // input_time is variable so we check that dimension 1 is different than its original value.
-
-        return (features.rank() == 3
-                && features.dim(2) == output_size()
-                && features.dim(1) != input_size());
-      }
-
       const StorageView* _upgraded_model;
 
     private:

include/ctranslate2/models/wavlm.h

@@ -54,8 +54,6 @@ namespace ctranslate2 {
     private:
       const std::shared_ptr<const WavLMModel> _model;
       const std::unique_ptr<layers::WavLMEncoder> _encoder;
-
-      StorageView maybe_encode(StorageView features);
     };
 
     class WavLM : public ReplicaPool<WavLMReplica> {

src/models/wavlm.cc

@@ -77,12 +77,7 @@ namespace ctranslate2 {
       features.move_to(device, dtype);
 
       StorageView encoder_output(dtype, device);
-      if (_encoder->_upgraded_model) {
-        encoder_output = maybe_encode(std::move(features));
-      }
-      else {
-        (*_encoder)(features, encoder_output);
-      }
+      (*_encoder)(features, encoder_output);
 
       if (to_cpu) {
         if (device != Device::CPU)
@@ -97,20 +92,6 @@ namespace ctranslate2 {
       return encoder_output;
     }
 
-    StorageView WavLMReplica::maybe_encode(StorageView features) {
-      const Device device = _model->device();
-      const DataType dtype = _encoder->output_type();
-
-      features.move_to(device, dtype);
-
-      if (_encoder->is_encoded(features))
-        return features;
-
-      StorageView encoder_output(dtype, device);
-      (*_encoder)(features, encoder_output);
-      return encoder_output;
-    }
-
     std::future<StorageView> WavLM::encode(const StorageView& features, const bool to_cpu) {
       return post<StorageView>(
         [features = features.sync_copy(), to_cpu](WavLMReplica& replica) mutable {