We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a8cc234 commit fb4be50Copy full SHA for fb4be50
swift/model/model_meta.py
@@ -87,12 +87,8 @@ def __post_init__(self):
87
self.loader = ModelLoader
88
if not isinstance(self.model_groups, (list, tuple)):
89
self.model_groups = [self.model_groups]
90
- self.candidate_templates = set([self.template])
91
- for model_group in self.model_groups:
92
- self.candidate_templates.add(model_group.template)
93
- self.candidate_templates.discard(None)
94
- self.candidate_templates = list(self.candidate_templates)
95
-
+ self.candidate_templates = list(
+ dict.fromkeys(t for t in [self.template] + [mg.template for mg in self.model_groups] if t is not None))
96
if self.model_type in MLLMModelType.__dict__:
97
self.is_multimodal = True
98
if self.model_type in RMModelType.__dict__:
0 commit comments