HugoLaurencon
commited on
Commit
·
0d17e60
1
Parent(s):
6514207
fix imports
Browse files- configuration_siglip.py +5 -5
- modeling_siglip.py +5 -5
configuration_siglip.py
CHANGED
@@ -20,12 +20,12 @@ from typing import TYPE_CHECKING, Any, Mapping, Optional, Union
|
|
20 |
|
21 |
|
22 |
if TYPE_CHECKING:
|
23 |
-
from
|
24 |
-
from
|
25 |
|
26 |
-
from
|
27 |
-
from
|
28 |
-
from
|
29 |
|
30 |
|
31 |
logger = logging.get_logger(__name__)
|
|
|
20 |
|
21 |
|
22 |
if TYPE_CHECKING:
|
23 |
+
from transformers.processing_utils import ProcessorMixin
|
24 |
+
from transformers.utils import TensorType
|
25 |
|
26 |
+
from transformers.configuration_utils import PretrainedConfig
|
27 |
+
from transformers.onnx import OnnxConfig
|
28 |
+
from transformers.utils import logging
|
29 |
|
30 |
|
31 |
logger = logging.get_logger(__name__)
|
modeling_siglip.py
CHANGED
@@ -22,17 +22,17 @@ import torch
|
|
22 |
import torch.utils.checkpoint
|
23 |
from torch import nn
|
24 |
|
25 |
-
from
|
26 |
-
from
|
27 |
-
from
|
28 |
-
from
|
29 |
ModelOutput,
|
30 |
add_start_docstrings,
|
31 |
add_start_docstrings_to_model_forward,
|
32 |
logging,
|
33 |
replace_return_docstrings,
|
34 |
)
|
35 |
-
from
|
36 |
|
37 |
|
38 |
logger = logging.get_logger(__name__)
|
|
|
22 |
import torch.utils.checkpoint
|
23 |
from torch import nn
|
24 |
|
25 |
+
from transformers.activations import ACT2FN
|
26 |
+
from transformers.modeling_outputs import BaseModelOutput, BaseModelOutputWithPooling
|
27 |
+
from transformers.modeling_utils import PreTrainedModel
|
28 |
+
from transformers.utils import (
|
29 |
ModelOutput,
|
30 |
add_start_docstrings,
|
31 |
add_start_docstrings_to_model_forward,
|
32 |
logging,
|
33 |
replace_return_docstrings,
|
34 |
)
|
35 |
+
from configuration_siglip import SiglipConfig, SiglipTextConfig, SiglipVisionConfig
|
36 |
|
37 |
|
38 |
logger = logging.get_logger(__name__)
|