Skip to content

Commit

Permalink
Update validate_models.py
Browse files Browse the repository at this point in the history
  • Loading branch information
NIXBLACK11 authored Nov 14, 2023
1 parent 18c1657 commit c26e775
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions laser_encoders/validate_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,10 +71,7 @@ def get_language_code(self, language_list: dict, lang: str) -> str:
def download_laser3(self, lang):
lang = self.get_language_code(LASER3_LANGUAGE, lang)
file_path = os.path.join(self.model_dir, f"laser3-{lang}.v1.pt")
if os.path.exists(file_path):
return False
else:
return True
return os.path.exists(file_path)

def download_laser2(self):
files = ["laser2.pt", "laser2.spm", "laser2.cvocab"]
Expand Down

0 comments on commit c26e775

Please sign in to comment.