admin commited on
Commit
3c85fae
·
1 Parent(s): f246c7c

fixed hf hub

Browse files
Files changed (2) hide show
  1. requirements.txt +2 -2
  2. utils.py +1 -5
requirements.txt CHANGED
@@ -10,7 +10,7 @@ matplotlib
10
  phonemizer
11
  tensorboard
12
  amfm_decompy
13
- transformers
14
  torch==2.3.1
15
  numpy==1.26.4
16
- librosa==0.9.1
 
 
10
  phonemizer
11
  tensorboard
12
  amfm_decompy
 
13
  torch==2.3.1
14
  numpy==1.26.4
15
+ librosa==0.9.1
16
+ huggingface-hub==0.28.1
utils.py CHANGED
@@ -18,11 +18,7 @@ logger = logging.getLogger(__name__)
18
 
19
  def load_checkpoint(checkpoint_path, model, optimizer=None, skip_optimizer=False):
20
  assert os.path.isfile(checkpoint_path)
21
- checkpoint_dict = torch.load(
22
- checkpoint_path,
23
- map_location="cpu",
24
- weights_only=False,
25
- )
26
  iteration = checkpoint_dict["iteration"]
27
  learning_rate = checkpoint_dict["learning_rate"]
28
  if (
 
18
 
19
  def load_checkpoint(checkpoint_path, model, optimizer=None, skip_optimizer=False):
20
  assert os.path.isfile(checkpoint_path)
21
+ checkpoint_dict = torch.load(checkpoint_path, map_location="cpu")
 
 
 
 
22
  iteration = checkpoint_dict["iteration"]
23
  learning_rate = checkpoint_dict["learning_rate"]
24
  if (