diff --git a/diffsynth/models/flux_text_encoder.py b/diffsynth/models/flux_text_encoder.py index 499b7be..3f34270 100644 --- a/diffsynth/models/flux_text_encoder.py +++ b/diffsynth/models/flux_text_encoder.py @@ -49,8 +49,8 @@ class FLUXTextEncoder2(T5EncoderModel): relative_attention_max_distance = 128, relative_attention_num_buckets = 32, tie_word_embeddings = False, - torch_dtype = bfloat16, # change - transformers_version = "4.43.3", # change + torch_dtype = "bfloat16", + transformers_version = "4.43.3", use_cache = True, vocab_size = 32128 ) @@ -115,4 +115,4 @@ class FLUXTextEncoder2StateDictConverter(): return state_dict_ def from_civitai(self, state_dict): - return self.from_diffusers(state_dict) \ No newline at end of file + return self.from_diffusers(state_dict)