This commit is contained in:
Yousef Rafat 2025-12-23 12:35:00 +02:00
parent e30298dda2
commit 5b0c80a093
6 changed files with 43 additions and 46 deletions

View File

@ -1,10 +1,8 @@
from dataclasses import dataclass
from typing import Optional, Tuple, Union, List, Dict, Any, Callable
import einops
from einops import rearrange, repeat
from einops import rearrange
import comfy.model_management
from torch import nn
import torch.nn.utils.rnn as rnn_utils
import torch.nn.functional as F
from math import ceil, pi
import torch
@ -15,7 +13,6 @@ from comfy.rmsnorm import RMSNorm
from torch.nn.modules.utils import _triple
from torch import nn
import math
import logging
class Cache:
def __init__(self, disable=False, prefix="", cache=None):