Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/transformers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -587,7 +587,6 @@
from .cache_utils import HQQQuantizedCache as HQQQuantizedCache
from .cache_utils import HQQQuantizedLayer as HQQQuantizedLayer
from .cache_utils import HybridCache as HybridCache
from .cache_utils import MambaCache as MambaCache
from .cache_utils import OffloadedCache as OffloadedCache
from .cache_utils import OffloadedStaticCache as OffloadedStaticCache
from .cache_utils import QuantizedCache as QuantizedCache
Expand Down Expand Up @@ -790,6 +789,7 @@
from .modeling_utils import AttentionInterface as AttentionInterface
from .modeling_utils import PreTrainedModel as PreTrainedModel
from .models import *
from .models.mamba.modeling_mamba import MambaCache as MambaCache
from .models.timm_wrapper import TimmWrapperImageProcessor as TimmWrapperImageProcessor

# Optimization
Expand Down
2 changes: 1 addition & 1 deletion src/transformers/models/timm_wrapper/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,8 @@

if TYPE_CHECKING:
from .configuration_timm_wrapper import *
from .image_processing_timm_wrapper import *
from .modeling_timm_wrapper import *
from .processing_timm_wrapper import *
else:
import sys

Expand Down