【深度学习】sdxl中的 text_encoder text_encoder_2 区别
if 'clip' in self.stages:
subfolder = 'text_encoder'
self.models['clip'] = CLIPModel(**models_args, embedding_dim=get_clip_embedding_dim(self.version, self.pipeline_type), output_hidden_states=self.config.get('clip_hidden_states', False), subfolder=subfolder)
if 'clip2' in self.stages:
subfolder = 'text_encoder_2'
self.models['clip2'] = CLIPWithProjModel(**models_args, output_hidden_states=self.config.get('clip_hidden_states', False), subfolder=subfolder)