OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-01 11:13:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

flux: clean up some dead code

Browse Source
This commit is contained in:
Won-Kyu Park 2024-09-28 21:03:35 +09:00
parent 4ad5f22c7b
commit 98cb284eb1
No known key found for this signature in database
GPG Key ID: 53AA79C8C9535D15
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/models/flux/flux.py
View File

@ -50,12 +50,6 @@ class Flux1ClipL(sd_hijack_clip.TextConditionalModel):
return self.tokenizer(texts, truncation=False, add_special_tokens=False)["input_ids"] return self.tokenizer(texts, truncation=False, add_special_tokens=False)["input_ids"]
def encode_with_transformers(self, tokens): def encode_with_transformers(self, tokens):
tokens_g = tokens.clone()
for batch_pos in range(tokens_g.shape[0]):
index = tokens_g[batch_pos].cpu().tolist().index(self.id_end)
tokens_g[batch_pos, index+1:tokens_g.shape[1]] = 0
l_out, l_pooled = self.clip_l(tokens) l_out, l_pooled = self.clip_l(tokens)
l_out = torch.cat([l_out], dim=-1) l_out = torch.cat([l_out], dim=-1)
l_out = torch.nn.functional.pad(l_out, (0, 4096 - l_out.shape[-1])) l_out = torch.nn.functional.pad(l_out, (0, 4096 - l_out.shape[-1]))