【深度学习】sdxl中的 text_encoder text_encoder_2 区别

时间:2024-01-27 19:07:46
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)