wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-06 11:32:31 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

..

Browse Source
This commit is contained in:
Yousef Rafat 2026-02-02 21:27:15 +02:00
parent 614b167994
commit f76e3a11b5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/trellis2/model.py
View File

@ -1,9 +1,9 @@
import torch
import torch.nn.functional as F
import torch.nn as nn
from vae import SparseTensor, SparseLinear, sparse_cat, VarLenTensor
from comfy.ldm.trellis2.vae import SparseTensor, SparseLinear, sparse_cat, VarLenTensor
from typing import Optional, Tuple, Literal, Union, List
from attention import sparse_windowed_scaled_dot_product_self_attention, sparse_scaled_dot_product_attention
from comfy.ldm.trellis2.attention import sparse_windowed_scaled_dot_product_self_attention, sparse_scaled_dot_product_attention
from comfy.ldm.genmo.joint_model.layers import TimestepEmbedder
class SparseGELU(nn.GELU):