Update validate_models.py

facebookresearch · Nov 14, 2023 · 3944556 · 3944556
1 parent 023eab2
commit 3944556
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/laser_encoders/validate_models.py b/laser_encoders/validate_models.py
@@ -80,11 +80,11 @@ def download_laser2(self):
         files = ["laser2.pt", "laser2.spm", "laser2.cvocab"]
         for file_name in files:
             file_path = os.path.join(self.model_dir, file_name)
-            if os.path.exists(file_path):
-                return False
-            else:
+            if not os.path.exists(file_path):
                 return True
 
+        return False
+
 
 CACHE_DIR = "/home/user/.cache/models"  # Change this to the desired cache directory