Update hunyuandit-mini-overfitting-flowmatching-dinol518-bf16-lr1e4-4096.yaml

repair
This commit is contained in:
s572915912
2025-08-06 01:12:13 +08:00
committed by GitHub
parent f4e0307665
commit d9fc4d31bf

View File

@@ -33,22 +33,6 @@ dataset:
std: &std [0.5, 0.5, 0.5]
#! Point cloud sampling
# Important: The total number of points (pc_size + pc_sharpedge_size) is coupled with the
# architecture of the pretrained ShapeVAE in `first_stage_config`. You must ensure that
# the total number of points provided by the dataset matches the model's expectation.
#
# For the default pretrained ShapeVAE (`tencent/Hunyuan3D-2.1` VAE), the model
# implicitly expects a total of 81920 points.
#
# Correct configuration (for the default model):
# pc_size: 81920
# pc_sharpedge_size: 0
#
# Incorrect configuration that will cause a `split_with_sizes` RuntimeError:
# pc_size: 10240
# pc_sharpedge_size: 10240
#
# If you need to use a different number of points, you must retrain the ShapeVAE model first.
pc_size: &pc_size 81920
pc_sharpedge_size: &pc_sharpedge_size 0
sharpedge_label: &sharpedge_label true
@@ -74,6 +58,11 @@ model:
first_stage_config:
target: hy3dshape.models.autoencoders.ShapeVAE
from_pretrained: tencent/Hunyuan3D-2.1
#Ensure same with the dataset setting
params:
pc_size: 81920
pc_sharpedge_size: 0
cond_stage_config:
target: hy3dshape.models.conditioner.SingleImageEncoder