Update modeling.py
Browse files- modeling.py +2 -1
modeling.py
CHANGED
|
@@ -4,7 +4,8 @@ import torch.nn as nn
|
|
| 4 |
from transformers import PreTrainedModel
|
| 5 |
import torch
|
| 6 |
# download repo
|
| 7 |
-
snapshot_download(repo_id="jadechoghari/VoiceRestore", repo_type="model")
|
|
|
|
| 8 |
from BigVGAN import bigvgan
|
| 9 |
from BigVGAN.meldataset import get_mel_spectrogram
|
| 10 |
from voice_restore import VoiceRestore
|
|
|
|
| 4 |
from transformers import PreTrainedModel
|
| 5 |
import torch
|
| 6 |
# download repo
|
| 7 |
+
repo_dir = snapshot_download(repo_id="jadechoghari/VoiceRestore", repo_type="model", local_dir="VoiceRestoreRepo")
|
| 8 |
+
sys.path.append(repo_dir)
|
| 9 |
from BigVGAN import bigvgan
|
| 10 |
from BigVGAN.meldataset import get_mel_spectrogram
|
| 11 |
from voice_restore import VoiceRestore
|