FlowProt / config.yaml
alibtsd's picture
Deploy FlowProt Docker Space
f34af6f verified
Raw
History Blame Contribute Delete
926 Bytes
# Runtime config for Hugging Face Space MVP.
# Canonical mode is unconditional inference.
app:
mvp_mode: unconditional
enable_conditional: false
enable_classifier: true
load_on_startup: false
device: auto
min_length: 32
max_length: 1024
max_samples_per_request: 4
inference:
seed: 123
output_dir: ./space_outputs
# Generative model checkpoint. Used unless overridden by FLOWPROT_CKPT_* env vars.
# ckpt_config_path is optional; defaults to <ckpt dir>/config.yaml.
ckpt_path: ./model/ckpt/epoch29.ckpt
ckpt_config_path: ./model/ckpt/config.yaml
classifier:
ckpt_path: ./model/ckpt/classifier_ckpt/epoch=90-step=728000.ckpt
guidance_scale: 0.2
target_class: 1
interpolant:
min_t: 1e-2
rots:
sample_schedule: exp
exp_rate: 10
trans:
sample_schedule: linear
sampling:
num_timesteps: 100
self_condition: true