diff --git a/requirements/required.txt b/requirements/required.txt index d948d981..8bf2f775 100644 --- a/requirements/required.txt +++ b/requirements/required.txt @@ -5,6 +5,6 @@ pretrainedmodels==0.7.4 safetensors==0.5.2 six==1.17.0 timm==1.0.14 -torch==2.5.1 -torchvision==0.20.1 +torch==2.6.0 +torchvision==0.21.0 tqdm==4.67.1 diff --git a/segmentation_models_pytorch/encoders/dpn.py b/segmentation_models_pytorch/encoders/dpn.py index 4fe84328..ffbd7198 100644 --- a/segmentation_models_pytorch/encoders/dpn.py +++ b/segmentation_models_pytorch/encoders/dpn.py @@ -34,7 +34,7 @@ class DPNEncoder(DPN, EncoderMixin): _is_torch_scriptable = False - _is_torch_exportable = False + _is_torch_exportable = True # since torch 2.6.0 def __init__( self,