From 90af5747946caad000f914e27b7aa0250133690d Mon Sep 17 00:00:00 2001 From: Xrvk Date: Sun, 18 Aug 2024 01:59:30 +0300 Subject: [PATCH] Remove unnecessary import and minor style change --- comfy/ldm/flux/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/comfy/ldm/flux/model.py b/comfy/ldm/flux/model.py index be475e97c..eb7767ca9 100644 --- a/comfy/ldm/flux/model.py +++ b/comfy/ldm/flux/model.py @@ -15,7 +15,6 @@ from .layers import ( ) from einops import rearrange, repeat -import numpy as np import comfy.ldm.common_dit @dataclass @@ -33,6 +32,7 @@ class FluxParams: qkv_bias: bool guidance_embed: bool + class Flux(nn.Module): """ Transformer model for flow matching on sequences. @@ -106,9 +106,9 @@ class Flux(nn.Module): if self.params.guidance_embed: if guidance is None: 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) ids = torch.cat((txt_ids, img_ids), dim=1)