CIS6270 / lecture_5 /examples /rectified.py
pranamanam's picture
Add Lecture 4 discrete diffusion and Lecture 5 discrete flow matching
c8293a4 verified
Raw
History Blame Contribute Delete
284 Bytes
"""Complete rectified training and generation example."""
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
from run import main
if __name__ == '__main__':
main(['--method', 'rectified', '--out', 'outputs/rectified'] + sys.argv[1:])