Spaces:

zzzzzeee
/

test_embedding_shape

Sleeping

zzzzzeee commited on 25 days ago

Commit

77b3a26

verified ·

1 Parent(s): 46ebef3

Update dinov2.py

Files changed (1) hide show

dinov2.py CHANGED Viewed

@@ -17,7 +17,7 @@ import torch.nn as nn
 import torch.utils.checkpoint
 from torch.nn.init import trunc_normal_
-from dinov2.layers import Mlp, PatchEmbed, SwiGLUFFNFused, MemEffAttention, NestedTensorBlock as Block
 logger = logging.getLogger("dinov2")

 import torch.utils.checkpoint
 from torch.nn.init import trunc_normal_
+from dinov2layers import Mlp, PatchEmbed, SwiGLUFFNFused, MemEffAttention, NestedTensorBlock as Block
 logger = logging.getLogger("dinov2")