xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2026-01-23 09:24:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Easymode 2025-02-19 00:37:29 +00:00 committed by GitHub
parent a9c6b5cbf0
commit faef87a80a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hy3dgen/shapegen/bpt/model/miche_conditioner.py
View File

@ -32,8 +32,8 @@ class PointConditioner(torch.nn.Module):
# open-source version of miche
if model_name == 'miche-256-feature':
ckpt_path = None
dir = os.path.dirname(os.path.abspath(__file__))
ckpt_path = None#os.path.join(dir, '..\shapevae-256.ckpt')
model_path = os.path.join(dir, '..\shapevae-256.yaml')
config_path = model_path

3
hy3dgen/shapegen/bpt/model/model.py
View File

@ -70,6 +70,7 @@ class MeshTransformer(Module):
conditioned_on_pc = True,
encoder_name = 'miche-256-feature',
encoder_freeze = False,
cond_dim = 768
):
super().__init__()
@ -107,7 +108,7 @@ class MeshTransformer(Module):
# load point_cloud encoder
if conditioned_on_pc:
print(f'Point cloud encoder: {encoder_name} | freeze: {encoder_freeze}')
self.conditioner = PointConditioner(model_name=encoder_name, freeze=encoder_freeze)
self.conditioner = PointConditioner(cond_dim=cond_dim, model_name=encoder_name, freeze=encoder_freeze)
cross_attn_dim_context = self.conditioner.dim_latent
else:
raise NotImplementedError