From d542b95ff3a33541e196af0c0f1a043fac009c41 Mon Sep 17 00:00:00 2001 From: jmisilo Date: Tue, 3 Oct 2023 15:00:12 +0200 Subject: [PATCH] fix: update model weights IDs in downloads.py file --- src/utils/downloads.py | 25 ++++++++++++++++--------- 1 file changed, 16 insertions(+), 9 deletions(-) diff --git a/src/utils/downloads.py b/src/utils/downloads.py index 6030f1c..9f1464e 100644 --- a/src/utils/downloads.py +++ b/src/utils/downloads.py @@ -1,14 +1,21 @@ -''' +""" Utility functions for loading weights. -''' +""" import gdown -def download_weights(checkpoint_fpath, model_size='L'): - ''' - Downloads weights from Google Drive. - ''' - download_id = '12h-NgryAf6zZdA1KclHdfzU35D1icjEp' if model_size.strip().upper() == 'L' else '1p91KBj-oUmuMfG2Gc33tEN5Js5HpV8YH' - - gdown.download(f'https://drive.google.com/uc?id={download_id}', checkpoint_fpath, quiet=False) \ No newline at end of file +def download_weights(checkpoint_fpath, model_size="L"): + """ + Downloads weights from Google Drive. + """ + + download_id = ( + "1pSQruQyg8KJq6VmzhMLFbT_VaHJMdlWF" + if model_size.strip().upper() == "L" + else "1Gh32arzhW06C1ZJyzcJSSfdJDi3RgWoG" + ) + + gdown.download( + f"https://drive.google.com/uc?id={download_id}", checkpoint_fpath, quiet=False + )