bytedancerneat commited on
Commit
499d83b
·
verified ·
1 Parent(s): 839b121

Update doubao_service.py

Browse files
Files changed (1) hide show
  1. doubao_service.py +3 -2
doubao_service.py CHANGED
@@ -3,6 +3,7 @@ import requests
3
  import json
4
  from volcenginesdkarkruntime import Ark
5
  from util.config_util import read_config as config
 
6
  from util import logger
7
  import volcenginesdkcore
8
  import volcenginesdkark
@@ -13,8 +14,8 @@ from util.logger_util import log_decorate
13
  class DouBaoService:
14
 
15
  def __init__(self, model_name):
16
- print(config())
17
- self.conf = config()[f"{model_name}ModelInfo"]
18
  self.client = self.init_client()
19
  self._complete_args = {}
20
 
 
3
  import json
4
  from volcenginesdkarkruntime import Ark
5
  from util.config_util import read_config as config
6
+ from util.config_util import load_json
7
  from util import logger
8
  import volcenginesdkcore
9
  import volcenginesdkark
 
14
  class DouBaoService:
15
 
16
  def __init__(self, model_name):
17
+ config = load_json('./conf/config.json')
18
+ self.conf = config[f"{model_name}ModelInfo"]
19
  self.client = self.init_client()
20
  self._complete_args = {}
21