diff --git a/python/ctranslate2/specs/wav2vec2_spec.py b/python/ctranslate2/specs/wav2vec2_spec.py
index 0cc6227d5..7b9b9cfe4 100644
--- a/python/ctranslate2/specs/wav2vec2_spec.py
+++ b/python/ctranslate2/specs/wav2vec2_spec.py
@@ -1,7 +1,6 @@
 from typing import List, Optional, Tuple
 
 import numpy as np
-import torch.nn as nn
 
 from ctranslate2.specs import common_spec, model_spec, transformer_spec