andreas/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Lint unused import (#5973)

Browse Source 
* Lint unused import

* nit

* Remove unused imports

* revert fix_torch import

* nit
This commit is contained in:
Chenlei Hu
2024-12-09 14:24:39 -06:00
committed by GitHub
parent e2fafe0686
commit 0fd4e6c778
40 changed files with 39 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/ldm/hydit/controlnet.py
View File

@@ -1,24 +1,17 @@
from typing import Any, Optional
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils import checkpoint
from comfy.ldm.modules.diffusionmodules.mmdit import (
Mlp,
TimestepEmbedder,
PatchEmbed,
RMSNorm,
)
from comfy.ldm.modules.diffusionmodules.util import timestep_embedding
from .poolers import AttentionPool
import comfy.latent_formats
from .models import HunYuanDiTBlock, calc_rope
from .posemb_layers import get_2d_rotary_pos_embed, get_fill_resize_and_crop
class HunYuanControlNet(nn.Module):

2
comfy/ldm/hydit/models.py
View File

@@ -1,8 +1,6 @@
from typing import Any
import torch
import torch.nn as nn
import torch.nn.functional as F
import comfy.ops
from comfy.ldm.modules.diffusionmodules.mmdit import Mlp, TimestepEmbedder, PatchEmbed, RMSNorm

1
comfy/ldm/hydit/poolers.py
View File

@@ -1,6 +1,5 @@
import torch
import torch.nn as nn
import torch.nn.functional as F
from comfy.ldm.modules.attention import optimized_attention
import comfy.ops