@ -1,12 +1,9 @@
from inspect import isfunction
import math
import torch
import torch.nn.functional as F
from torch import nn, einsum
from einops import rearrange, repeat
from typing import Optional, Any
from functools import partial
from .diffusionmodules.util import checkpoint, AlphaBlender, timestep_embedding
from .sub_quadratic_attention import efficient_dot_product_attention
from abc import abstractmethod
import numpy as np
import torch as th
import torch.nn as nn
from einops import rearrange
from .util import (
checkpoint,
@ -1,13 +1,9 @@
from .k_diffusion import sampling as k_diffusion_sampling
from .extra_samplers import uni_pc
import enum
import collections
from comfy import model_management
from comfy import model_base
import comfy.utils
import comfy.conds
def get_area_and_mult(conds, x_in, timestep_in):
area = (x_in.shape[2], x_in.shape[3], 0, 0)