mr2along commited on
Commit
c3f58dd
·
verified ·
1 Parent(s): 63b5c41

Update bark_infer_cpu.py

Browse files
Files changed (1) hide show
  1. 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