nameissakthi's picture
Add model architecture code
27871e7
# Inference and generation components