From 7d2467e830ca8f6dfab67e74867aa08fded3d75e Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sat, 5 Oct 2024 13:22:39 -0400 Subject: [PATCH] Some minor cleanups. --- comfy/clip_vision.py | 3 +-- comfy/float.py | 1 - comfy/sd.py | 1 - 3 files changed, 1 insertion(+), 4 deletions(-) diff --git a/comfy/clip_vision.py b/comfy/clip_vision.py index 20dc3345..64392e27 100644 --- a/comfy/clip_vision.py +++ b/comfy/clip_vision.py @@ -109,8 +109,7 @@ def load_clipvision_from_sd(sd, prefix="", convert_keys=False): keys = list(sd.keys()) for k in keys: if k not in u: - t = sd.pop(k) - del t + sd.pop(k) return clip def load(ckpt_path): diff --git a/comfy/float.py b/comfy/float.py index 4a6ae677..0c8f1d6c 100644 --- a/comfy/float.py +++ b/comfy/float.py @@ -1,5 +1,4 @@ import torch -import math def calc_mantissa(abs_x, exponent, normal_mask, MANTISSA_BITS, EXPONENT_BIAS, generator=None): mantissa_scaled = torch.where( diff --git a/comfy/sd.py b/comfy/sd.py index 3b4cff74..a494e531 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -29,7 +29,6 @@ import comfy.text_encoders.long_clipl import comfy.model_patcher import comfy.lora import comfy.t2i_adapter.adapter -import comfy.supported_models_base import comfy.taesd.taesd def load_lora_for_models(model, clip, lora, strength_model, strength_clip):