Update model.py
Browse files
model.py
CHANGED
|
@@ -18,6 +18,7 @@ import soundfile as sf
|
|
| 18 |
import io
|
| 19 |
from pydub import AudioSegment, silence
|
| 20 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 21 |
import os
|
| 22 |
|
| 23 |
class INF5Config(PretrainedConfig):
|
|
|
|
| 18 |
import io
|
| 19 |
from pydub import AudioSegment, silence
|
| 20 |
from huggingface_hub import hf_hub_download
|
| 21 |
+
from safetensors.torch import load_file
|
| 22 |
import os
|
| 23 |
|
| 24 |
class INF5Config(PretrainedConfig):
|