Beispiel #1
0
 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)
Beispiel #2
0
 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)
Beispiel #3
0
 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)