export CUDA_VISIBLE_DEVICES=6

python ./smiles/generation.py \
    --checkpoint_path ./smiles/ckpt/SMILESReDi_v1.ckpt \
    --n_samples 400 \
    --n_steps 16 \
    --temperature 1 \
    --seq_len 200
