Update src/music2cocktailrep/training/latent_translation/setup_trained_model.py

src/music2cocktailrep/training/latent_translation/setup_trained_model.py

@@ -7,6 +7,7 @@ from src.cocktails.utilities.cocktail_utilities import get_bunch_of_rep_keys
 import os
 from huggingface_hub import hf_hub_download
 from shutil import copy
+import hashlib
 
 TOKEN = os.environ['token']
 rep_keys = get_bunch_of_rep_keys()['custom']
@@ -33,6 +34,7 @@ def setup_trained_model(checkpoint_path=TRANSLATION_VAE_CHKP_PATH):
                                nb_classes=params['nb_classes'],
                                dropout=params['dropout'])
     print('HEREEE: ', torch.sum(torch.Tensor([param.sum() for param in list(model.parameters())])))
+    print('model hash: ', hashlib.md5(open(model_path, 'rb').read()).hexdigest())
     stats = params['stats']
     stats_music = np.array(stats['mean_std_music_rep'])
     stats_cocktail = np.array(stats['mean_std_cocktail_rep_norm11'])