theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-20 23:58:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

update image encoder

Browse Source
This commit is contained in:
Artiprocher
2024-01-30 14:08:30 +08:00
parent e076e66827
commit 40d17569d3
2 changed files with 90 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/models/sd_text_encoder.py
View File

@@ -16,7 +16,7 @@ class CLIPEncoderLayer(torch.nn.Module):
def quickGELU(self, x):
return x * torch.sigmoid(1.702 * x)
def forward(self, hidden_states, attn_mask):
def forward(self, hidden_states, attn_mask=None):
residual = hidden_states
hidden_states = self.layer_norm1(hidden_states)