hugging-apps/echo-memory

1import torch2from transformers import T5EncoderModel, T5Config3from .sd_text_encoder import SDTextEncoder4 5 6 7class FluxTextEncoder2(T5EncoderModel):8    def __init__(self, config):9        super().__init__(config)10        self.eval()11 12    def forward(self, input_ids):13        outputs = super().forward(input_ids=input_ids)14        prompt_emb = outputs.last_hidden_state15        return prompt_emb16 17    @staticmethod18    def state_dict_converter():19        return FluxTextEncoder2StateDictConverter()20 21 22 23class FluxTextEncoder2StateDictConverter():24    def __init__(self):25        pass26 27    def from_diffusers(self, state_dict):28        state_dict_ = state_dict29        return state_dict_30 31    def from_civitai(self, state_dict):32        return self.from_diffusers(state_dict)33