diff --git a/utils/models/cogagent_model.py b/utils/models/cogagent_model.py index fc4967d0..4c104fe6 100644 --- a/utils/models/cogagent_model.py +++ b/utils/models/cogagent_model.py @@ -36,7 +36,9 @@ def forward(self, x): from .eva_clip_model import EVA2CLIPModel import argparse from copy import deepcopy -def override_dist_dtype_device_args(args, b={}): +def override_dist_dtype_device_args(args, b=None): + if b is None: + b = {} if args.mode == 'inference': minimal_args = argparse.Namespace( world_size=args.world_size, diff --git a/utils/models/cogvlm_model.py b/utils/models/cogvlm_model.py index 30600eb4..2f109111 100644 --- a/utils/models/cogvlm_model.py +++ b/utils/models/cogvlm_model.py @@ -35,7 +35,9 @@ def forward(self, x): from .eva_clip_model import EVA2CLIPModel import argparse from copy import deepcopy -def override_dist_dtype_device_args(args, b={}): +def override_dist_dtype_device_args(args, b=None): + if b is None: + b = {} if args.mode == 'inference': minimal_args = argparse.Namespace( world_size=args.world_size,