Remove some unused imports.

comfyanonymous · comfyanonymous · commit 0920e0e5febc · 2024-05-27T19:08:27.000-04:00
diff --git a/comfy/ldm/cascade/stage_b.py b/comfy/ldm/cascade/stage_b.py
@@ -17,7 +17,6 @@
 """
 
 import math
-import numpy as np
 import torch
 from torch import nn
 from .common import AttnBlock, LayerNorm2d_op, ResBlock, FeedForwardBlock, TimestepBlock
diff --git a/comfy/ldm/cascade/stage_c.py b/comfy/ldm/cascade/stage_c.py
@@ -18,7 +18,6 @@
 
 import torch
 from torch import nn
-import numpy as np
 import math
 from .common import AttnBlock, LayerNorm2d_op, ResBlock, FeedForwardBlock, TimestepBlock
 # from .controlnet import ControlNetDeliverer
diff --git a/comfy/ldm/models/autoencoder.py b/comfy/ldm/models/autoencoder.py
@@ -1,6 +1,4 @@
 import torch
-# import pytorch_lightning as pl
-import torch.nn.functional as F
 from contextlib import contextmanager
 from typing import Any, Dict, List, Optional, Tuple, Union
 
diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py
@@ -3,7 +3,7 @@
 import torch.nn.functional as F
 from torch import nn, einsum
 from einops import rearrange, repeat
-from typing import Optional, Any
+from typing import Optional
 import logging
 
 from .diffusionmodules.util import AlphaBlender, timestep_embedding
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -2,7 +2,6 @@
 import logging
 from enum import Enum
 from comfy.cli_args import args
-import comfy.utils
 import torch
 import sys
 import platform
diff --git a/comfy/sd.py b/comfy/sd.py
@@ -14,7 +14,6 @@
 from . import clip_vision
 from . import gligen
 from . import diffusers_convert
-from . import model_base
 from . import model_detection
 
 from . import sd1_clip
diff --git a/comfy/sd2_clip.py b/comfy/sd2_clip.py
@@ -1,5 +1,4 @@
 from comfy import sd1_clip
-import torch
 import os
 
 class SD2ClipHModel(sd1_clip.SDClipModel):
diff --git a/comfy_extras/nodes_canny.py b/comfy_extras/nodes_canny.py
@@ -1,10 +1,5 @@
-import math
-
-import torch
-import torch.nn.functional as F
-import comfy.model_management
-
 from kornia.filters import canny
+import comfy.model_management
 
 
 class Canny:
diff --git a/comfy_extras/nodes_sag.py b/comfy_extras/nodes_sag.py
@@ -4,7 +4,6 @@
 import math
 
 from einops import rearrange, repeat
-import os
 from comfy.ldm.modules.attention import optimized_attention
 import comfy.samplers
 
diff --git a/comfy_extras/nodes_sdupscale.py b/comfy_extras/nodes_sdupscale.py
@@ -1,5 +1,4 @@
 import torch
-import nodes
 import comfy.utils
 
 class SD_4XUpscale_Conditioning:

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,4 @@`
`1`	`1`	`from comfy import sd1_clip`
`2`		`-import torch`
`3`	`2`	`import os`
`4`	`3`
`5`	`4`	`class SD2ClipHModel(sd1_clip.SDClipModel):`
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,4 @@`
`1`	`1`	`import torch`
`2`		`-import nodes`
`3`	`2`	`import comfy.utils`
`4`	`3`
`5`	`4`	`class SD_4XUpscale_Conditioning:`