nvidia
/

C-RADIOv3-L

Model card Files Files and versions Community

mranzinger commited on 28 days ago

Commit

4a3af8c

·

verified ·

1 Parent(s): 9b715d4

Update enable_cpe_support.py

Files changed (1) hide show

enable_cpe_support.py +1 -1

enable_cpe_support.py CHANGED Viewed

@@ -14,7 +14,7 @@ from torch import nn
 from timm.models import VisionTransformer, checkpoint_seq
-from radio.feature_normalizer import IntermediateFeatureNormalizerBase, NullIntermediateFeatureNormalizer
 from .extra_models import DinoWrapper
 from .vit_patch_generator import ViTPatchGenerator

 from timm.models import VisionTransformer, checkpoint_seq
+from .feature_normalizer import IntermediateFeatureNormalizerBase, NullIntermediateFeatureNormalizer
 from .extra_models import DinoWrapper
 from .vit_patch_generator import ViTPatchGenerator