MatterGen / demo /configs /generate_base.yaml
dadadaxi's picture
Upload folder using huggingface_hub
f15d29e verified
Raw
History Blame Contribute Delete
251 Bytes
# 基础模型单样本生成。修改 checkpoint、output、batch_size 和 num_batches。
task: generate
args:
checkpoint: ${ONESCIENCE_MODELS_DIR}/mattergen/mattergen_base
output: outputs/generate/mattergen_base
batch_size: 1
num_batches: 1