chong.zhang commited on
Commit
eec9b94
·
1 Parent(s): 19e7dfa
inspiremusic/cli/frontend.py CHANGED
@@ -12,14 +12,8 @@
12
  # See the License for the specific language governing permissions and
13
  # limitations under the License.
14
  from functools import partial
15
- import onnxruntime
16
  import torch
17
- import numpy as np
18
- import whisper
19
  from typing import Callable
20
- import torchaudio.compliance.kaldi as kaldi
21
- import torchaudio
22
- import os
23
  import re
24
  import inflect
25
  from inspiremusic.cli.model import InspireMusicModel
 
12
  # See the License for the specific language governing permissions and
13
  # limitations under the License.
14
  from functools import partial
 
15
  import torch
 
 
16
  from typing import Callable
 
 
 
17
  import re
18
  import inflect
19
  from inspiremusic.cli.model import InspireMusicModel
inspiremusic/cli/model.py CHANGED
@@ -11,13 +11,11 @@
11
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
  # See the License for the specific language governing permissions and
13
  # limitations under the License.
14
- import torch
15
  import numpy as np
16
  import threading
17
  import time
18
  from contextlib import nullcontext
19
  import uuid
20
- from inspiremusic.utils.common import fade_in_out
21
  from inspiremusic.music_tokenizer.vqvae import VQVAE
22
  from inspiremusic.wavtokenizer.decoder.pretrained import WavTokenizer
23
  from torch.cuda.amp import autocast
 
11
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
  # See the License for the specific language governing permissions and
13
  # limitations under the License.
 
14
  import numpy as np
15
  import threading
16
  import time
17
  from contextlib import nullcontext
18
  import uuid
 
19
  from inspiremusic.music_tokenizer.vqvae import VQVAE
20
  from inspiremusic.wavtokenizer.decoder.pretrained import WavTokenizer
21
  from torch.cuda.amp import autocast