From e43874b483facfde0d33ca2a6d2f9d56ba0855ed Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 25 Jan 2025 20:32:44 +0200
Subject: [PATCH] Update vae.py

---
 hy3dgen/shapegen/models/vae.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/hy3dgen/shapegen/models/vae.py b/hy3dgen/shapegen/models/vae.py
index e1761ac..03a26a0 100755
--- a/hy3dgen/shapegen/models/vae.py
+++ b/hy3dgen/shapegen/models/vae.py
@@ -31,7 +31,10 @@ import torch.nn.functional as F
 from einops import rearrange, repeat
 from skimage import measure
 from tqdm import tqdm
-from sageattention import sageattn
+try:
+    from sageattention import sageattn
+except ImportError:
+    sageattn = None
 from comfy.utils import ProgressBar
 
 class FourierEmbedder(nn.Module):