xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2026-05-02 02:35:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update vae.py

Browse Source
This commit is contained in:
kijai 2025-01-25 20:32:44 +02:00
parent a86f12236e
commit e43874b483
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
hy3dgen/shapegen/models/vae.py
View File

@ -31,7 +31,10 @@ import torch.nn.functional as F
from einops import rearrange, repeat from einops import rearrange, repeat
from skimage import measure from skimage import measure
from tqdm import tqdm from tqdm import tqdm
from sageattention import sageattn try:
from sageattention import sageattn
except ImportError:
sageattn = None
from comfy.utils import ProgressBar from comfy.utils import ProgressBar
class FourierEmbedder(nn.Module): class FourierEmbedder(nn.Module):