xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 11:46:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Rollback minor change

Browse Source
This commit is contained in:
eesahe 2024-08-17 19:04:59 +00:00
parent b6ad02282a
commit 7e7e40f7e3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/flux/model.py
View File

@ -133,9 +133,9 @@ class Flux(nn.Module):
if self.params.guidance_embed: if self.params.guidance_embed:
if guidance is None: if guidance is None:
raise ValueError("Didn't get guidance strength for guidance distilled model.") raise ValueError("Didn't get guidance strength for guidance distilled model.")
vec.add_(self.guidance_in(timestep_embedding(guidance, 256).to(img.dtype))) vec = vec + self.guidance_in(timestep_embedding(guidance, 256).to(img.dtype))
vec.add_(self.vector_in(y)) vec = vec + self.vector_in(y)
txt = self.txt_in(txt) txt = self.txt_in(txt)
ids = torch.cat((txt_ids, img_ids), dim=1) ids = torch.cat((txt_ids, img_ids), dim=1)