theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 18:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

add flux_text_encoder.py

Browse Source
This commit is contained in:
Qianyi Zhao
2024-08-13 22:26:10 -05:00
committed by GitHub
parent abe88c899e
commit d1ac96c1ab
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/models/flux_text_encoder.py
View File

@@ -49,8 +49,8 @@ class FLUXTextEncoder2(T5EncoderModel):
relative_attention_max_distance = 128, relative_attention_max_distance = 128,
relative_attention_num_buckets = 32, relative_attention_num_buckets = 32,
tie_word_embeddings = False, tie_word_embeddings = False,
torch_dtype = bfloat16, # change torch_dtype = "bfloat16",
transformers_version = "4.43.3", # change transformers_version = "4.43.3",
use_cache = True, use_cache = True,
vocab_size = 32128 vocab_size = 32128
) )