File tree Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Original file line number Diff line number Diff line change 153153 "FD_MODEL_USE_SAFETENSORS" : lambda : int (os .getenv ("FD_MODEL_USE_SAFETENSORS" , "1" )) == 1 ,
154154 "FD_MODEL_USE_OFFLINE_QUANT" : lambda : int (os .getenv ("FD_MODEL_USE_OFFLINE_QUANT" , "1" )) == 1 ,
155155 "FD_MOE_QUANT_TYPE" : lambda : os .getenv ("FD_MOE_QUANT_TYPE" , "w4a8" ),
156+ # The AK of bos storing the features while multi_modal infer
156157 "ENCODE_FEATURE_BOS_AK" : lambda : os .getenv ("ENCODE_FEATURE_BOS_AK" ),
158+ # The SK of bos storing the features while multi_modal infer
157159 "ENCODE_FEATURE_BOS_SK" : lambda : os .getenv ("ENCODE_FEATURE_BOS_SK" ),
160+ # The ENDPOINT of bos storing the features while multi_modal infer
161+ "ENCODE_FEATURE_ENDPOINT" : lambda : os .getenv ("ENCODE_FEATURE_ENDPOINT" ),
158162 # Enable offline perf test mode for PD disaggregation
159163 "FD_OFFLINE_PERF_TEST_FOR_PD" : lambda : int (os .getenv ("FD_OFFLINE_PERF_TEST_FOR_PD" , "0" )),
160164 "FD_ENABLE_E2W_TENSOR_CONVERT" : lambda : int (os .getenv ("FD_ENABLE_E2W_TENSOR_CONVERT" , "0" )),
Original file line number Diff line number Diff line change @@ -972,7 +972,8 @@ def init_bos_client():
972972 from baidubce .services .bos .bos_client import BosClient
973973
974974 cfg = BceClientConfiguration (
975- credentials = BceCredentials (envs .ENCODE_FEATURE_BOS_AK , envs .ENCODE_FEATURE_BOS_SK ), endpoint = "bj.bcebos.com"
975+ credentials = BceCredentials (envs .ENCODE_FEATURE_BOS_AK , envs .ENCODE_FEATURE_BOS_SK ),
976+ endpoint = envs .ENCODE_FEATURE_ENDPOINT ,
976977 )
977978 return BosClient (cfg )
978979
You can’t perform that action at this time.
0 commit comments