add requirements

condition/midas/midas/vit.py

@@ -383,22 +383,39 @@ def _make_vit_b_rn50_backbone(
     pretrained = nn.Module()
 
     pretrained.model = model
+    activations = []
 
     if use_vit_only == True:
         pretrained.model.blocks[hooks[0]].register_forward_hook(
-            get_activation("1"))
+            hook_act)
         pretrained.model.blocks[hooks[1]].register_forward_hook(
-            get_activation("2"))
+            hook_act)
     else:
         pretrained.model.patch_embed.backbone.stages[0].register_forward_hook(
-            get_activation("1"))
+            hook_act)
         pretrained.model.patch_embed.backbone.stages[1].register_forward_hook(
-            get_activation("2"))
+            hook_act)
 
     pretrained.model.blocks[hooks[2]].register_forward_hook(
-        get_activation("3"))
+        hook_act)
     pretrained.model.blocks[hooks[3]].register_forward_hook(
-        get_activation("4"))
+       hook_act)
+
+    # if use_vit_only == True:
+    #     pretrained.model.blocks[hooks[0]].register_forward_hook(
+    #         get_activation("1"))
+    #     pretrained.model.blocks[hooks[1]].register_forward_hook(
+    #         get_activation("2"))
+    # else:
+    #     pretrained.model.patch_embed.backbone.stages[0].register_forward_hook(
+    #         get_activation("1"))
+    #     pretrained.model.patch_embed.backbone.stages[1].register_forward_hook(
+    #         get_activation("2"))
+
+    # pretrained.model.blocks[hooks[2]].register_forward_hook(
+    #     get_activation("3"))
+    # pretrained.model.blocks[hooks[3]].register_forward_hook(
+    #     get_activation("4"))
 
     pretrained.activations = activations