multimodalart HF Staff commited on
Commit
19da0cf
·
1 Parent(s): f9b43f4

[Admin maintenance] Support new ZeroGPU hardware (#6)

Browse files

- [Admin maintenance] Support new ZeroGPU hardware (ce08b979912fafcf0d8e9656345407c500a189d2)

Files changed (3) hide show
  1. README.md +1 -1
  2. demos/melodyflow_app.py +4 -0
  3. requirements.txt +2 -2
README.md CHANGED
@@ -10,7 +10,7 @@ emoji: 🎵
10
  colorFrom: gray
11
  colorTo: blue
12
  sdk: gradio
13
- sdk_version: 4.44.1
14
  pinned: true
15
  license: cc-by-nc-4.0
16
  disable_embedding: true
 
10
  colorFrom: gray
11
  colorTo: blue
12
  sdk: gradio
13
+ sdk_version: 5.49.1
14
  pinned: true
15
  license: cc-by-nc-4.0
16
  disable_embedding: true
demos/melodyflow_app.py CHANGED
@@ -17,6 +17,10 @@ import typing as tp
17
  import warnings
18
 
19
  import torch
 
 
 
 
20
  import gradio as gr
21
 
22
  from audiocraft.data.audio_utils import convert_audio
 
17
  import warnings
18
 
19
  import torch
20
+ # torch.load weights_only flipped to True in 2.6; audiocraft checkpoints
21
+ # contain omegaconf DictConfig globals — restore the old default.
22
+ _orig_torch_load = torch.load
23
+ torch.load = lambda *a, **k: _orig_torch_load(*a, **{**k, "weights_only": k.get("weights_only", False)})
24
  import gradio as gr
25
 
26
  from audiocraft.data.audio_utils import convert_audio
requirements.txt CHANGED
@@ -9,8 +9,8 @@ num2words
9
  numpy
10
  sentencepiece
11
  spacy>=3.6.1
12
- torch==2.4.0
13
- torchaudio
14
  huggingface_hub
15
  tqdm
16
  transformers>=4.31.0 # need Encodec there.
 
9
  numpy
10
  sentencepiece
11
  spacy>=3.6.1
12
+ torch==2.8.0
13
+ torchaudio==2.8.0
14
  huggingface_hub
15
  tqdm
16
  transformers>=4.31.0 # need Encodec there.