Spaces:
Running
Running
admin
commited on
Commit
·
3c85fae
1
Parent(s):
f246c7c
fixed hf hub
Browse files- requirements.txt +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 (
|