From 2bb41bf75274bd4ae09f67c3ff69cdf5e9bf507c Mon Sep 17 00:00:00 2001
From: Jiayu Liu <jiayu@hey.com>
Date: Tue, 20 Feb 2024 10:35:06 +0800
Subject: [PATCH] more config changes

---
 src/detection/mod.rs       | 2 ++
 src/detection/segformer.rs | 1 +
 src/recognition/mod.rs     | 1 -
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/detection/mod.rs b/src/detection/mod.rs
index 78a857b..cc9a15a 100644
--- a/src/detection/mod.rs
+++ b/src/detection/mod.rs
@@ -1,3 +1,5 @@
+//! Detection module, consisting of segformer implementation
+
 mod segformer;
 
 use crate::error::Result;
diff --git a/src/detection/segformer.rs b/src/detection/segformer.rs
index b91e598..45e6fc3 100644
--- a/src/detection/segformer.rs
+++ b/src/detection/segformer.rs
@@ -1,3 +1,4 @@
+//! Segformer implementation
 use candle_core::{Module, ModuleT, Result, Tensor, D};
 use candle_nn::{
     conv2d, conv2d_no_bias, layer_norm, linear, Activation, Conv2d, Conv2dConfig, Linear,
diff --git a/src/recognition/mod.rs b/src/recognition/mod.rs
index 090a510..f28b3f9 100644
--- a/src/recognition/mod.rs
+++ b/src/recognition/mod.rs
@@ -16,7 +16,6 @@ use swin_transformer::SwinModel;
 pub struct Config {
     encoder: SwinConfig,
     decoder: MBartConfig,
-    pad_token_id: i32,
 }
 
 #[derive(Debug, Clone)]