ModuFlow/mask-ddpm
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
MingzheYang
2026-01-24 01:19:13 +08:00
parent 444ecd856b
commit 0105a0b206
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/export_samples.py
View File

@@ -112,8 +112,6 @@ def main():
int_like = stats.get("int_like", {}) int_like = stats.get("int_like", {})
max_decimals = stats.get("max_decimals", {}) max_decimals = stats.get("max_decimals", {})
transforms = stats.get("transform", {}) transforms = stats.get("transform", {})
cont_target = str(cfg.get("cont_target", "eps"))
cont_clamp_x0 = float(cfg.get("cont_clamp_x0", 0.0))
vocab_json = json.load(open(args.vocab_path, "r", encoding="utf-8")) vocab_json = json.load(open(args.vocab_path, "r", encoding="utf-8"))
vocab = vocab_json["vocab"] vocab = vocab_json["vocab"]
@@ -140,6 +138,8 @@ def main():
cond_vocab_size = len(sorted(base.glob(pat))) cond_vocab_size = len(sorted(base.glob(pat)))
if cond_vocab_size <= 0: if cond_vocab_size <= 0:
raise SystemExit("use_condition enabled but no files matched data_glob: %s" % cfg_glob) raise SystemExit("use_condition enabled but no files matched data_glob: %s" % cfg_glob)
cont_target = str(cfg.get("cont_target", "eps"))
cont_clamp_x0 = float(cfg.get("cont_clamp_x0", 0.0))
model = HybridDiffusionModel( model = HybridDiffusionModel(
cont_dim=len(cont_cols), cont_dim=len(cont_cols),