fix wrong apex help message

apex cannot be installed by pip
ngoiyaeric · Jan 4, 2024 · 6c7594d · 6c7594d
1 parent 390b7c6
commit 6c7594d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/models/eva_clip_L_hf.py b/utils/models/eva_clip_L_hf.py
@@ -679,7 +679,7 @@ def forward(self, x: torch.Tensor):
     from apex.normalization import FusedLayerNorm
 except:
     FusedLayerNorm = LayerNorm
-    print("Please 'pip install apex'")
+    print("Please build and install Nvidia apex package with option '--cuda_ext' according to https://github.com/NVIDIA/apex#from-source .")
 
 
 @dataclass
@@ -787,4 +787,4 @@ def __init__(self, config):
 
     def forward(self, images):
         enc = self.vit(images)
-        return enc + self.pos_embed.unsqueeze(0)
+        return enc + self.pos_embed.unsqueeze(0)