Lint unused import (#5973)

* 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

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):