From d1ac96c1abf48b25bf9cef34e59091d947a3fd36 Mon Sep 17 00:00:00 2001
From: Qianyi Zhao <49068354+Qing112@users.noreply.github.com>
Date: Tue, 13 Aug 2024 22:26:10 -0500
Subject: [PATCH] add flux_text_encoder.py

---
 diffsynth/models/flux_text_encoder.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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)