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