Update modeling_divedoc.py

modeling_divedoc.py

@@ -9,7 +9,7 @@ import torch
 import torch.nn as nn
 import torch.utils.checkpoint
 import torch.nn.functional as F
-
+from torch import Tensor
 
 from transformers import Cache, HybridCache, StaticCache
 from transformers.modeling_outputs import BaseModelOutput
@@ -21,7 +21,7 @@ from transformers.models.paligemma.configuration_paligemma import PaliGemmaConfi
 from transformers.models.donut.modeling_donut_swin import DonutSwinModel
 
 
-from .config_divedoc import SwinPamVisionEncoderConfig, SiglipPAMVisionEncoderConfig, DIVEdocConfig
+from .configuration_divedoc import SwinPamVisionEncoderConfig, SiglipPAMVisionEncoderConfig, DIVEdocConfig
 from typing import List, Optional, Tuple, Union, Literal
 from dataclasses import dataclass