def _get_shared_models(self, args: "DictConfig") -> Dict[str, dict]: _ = args # Not needed models_needed = list(self.conversations_needed.keys()) active_models = [ m for m in models_needed if self.conversations_needed[m] > 0 ] return TurkLikeAgent.get_bot_agents(args=args, active_models=active_models)
def _get_shared_models(self, args: "DictConfig") -> Dict[str, dict]: with open(args.blueprint.model_opt_path) as f: all_model_opts = yaml.safe_load(f.read()) active_model_opts = { model: opt for model, opt in all_model_opts.items() if self.conversations_needed[model] > 0 } return TurkLikeAgent.get_bot_agents(args=args, model_opts=active_model_opts)
def _get_shared_models(self, args: "DictConfig") -> Dict[str, dict]: with open(args.blueprint.model_opt_path) as f: model_opts = yaml.safe_load(f.read()) return TurkLikeAgent.get_bot_agents(args=args, model_opts=model_opts)