config.py

# Checkpoints & Dataset root
TROL_1_8B="BK-Lee/TroL-1.8B"
TROL_3_8B="BK-Lee/TroL-3.8B"
TROL_7B="BK-Lee/TroL-7B"