use our blip

2026-08-16 13:13:16 +02:00 · 2023-12-12 21:07:39 -08:00
parent c175afb394
commit 322aa5a724
10 changed files with 25 additions and 31 deletions
@@ -18,7 +18,10 @@ from timm.models.registry import register_model
 from timm.models.layers import trunc_normal_, DropPath
 from timm.models.helpers import named_apply, adapt_input_conv

-from fairscale.nn.checkpoint.checkpoint_activations import checkpoint_wrapper
+
+def checkpoint_wrapper(x):
+    return x
+

 class Mlp(nn.Module):
    """ MLP as used in Vision Transformer, MLP-Mixer and related networks