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