Spaces:

zzzzzeee
/

test_embedding_shape

Sleeping

zzzzzeee commited on 22 days ago

Commit

4212ed7

verified ·

1 Parent(s): c7674fb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from PIL import Image
 import torchvision.transforms as transforms
 from functools import partial
 from dinov2 import DinoVisionTransformer
-from dinov2models import Block, MemEffAttention
 # 定义模型（使用 vit_small 作为示例）
 def create_model(height, width):

 import torchvision.transforms as transforms
 from functools import partial
 from dinov2 import DinoVisionTransformer
+from dinov2layers import Block, MemEffAttention
 # 定义模型（使用 vit_small 作为示例）
 def create_model(height, width):