ccolas commited on
Commit
62796c8
1 Parent(s): 2300540

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

Browse files
src/music2cocktailrep/training/latent_translation/setup_trained_model.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  from huggingface_hub import hf_hub_download
9
  from shutil import copy
10
 
11
- TOKEN = os.environ['secret']
12
  rep_keys = get_bunch_of_rep_keys()['custom']
13
 
14
  def setup_trained_model(checkpoint_path=TRANSLATION_VAE_CHKP_PATH):
 
8
  from huggingface_hub import hf_hub_download
9
  from shutil import copy
10
 
11
+ TOKEN = os.environ['token']
12
  rep_keys = get_bunch_of_rep_keys()['custom']
13
 
14
  def setup_trained_model(checkpoint_path=TRANSLATION_VAE_CHKP_PATH):