Spaces:
Running
Running
Update bark_infer_cpu.py
Browse files- bark_infer_cpu.py +2 -0
bark_infer_cpu.py
CHANGED
@@ -7,6 +7,8 @@ import os
|
|
7 |
|
8 |
# Tắt cảnh báo
|
9 |
os.environ['BARK_IGNORE_WARNINGS'] = '1'
|
|
|
|
|
10 |
|
11 |
from bark.generation import preload_models
|
12 |
import torch.serialization
|
|
|
7 |
|
8 |
# Tắt cảnh báo
|
9 |
os.environ['BARK_IGNORE_WARNINGS'] = '1'
|
10 |
+
# Cho phép global object bị block
|
11 |
+
torch.serialization.add_safe_globals([np.core.multiarray.scalar])
|
12 |
|
13 |
from bark.generation import preload_models
|
14 |
import torch.serialization
|