Remove optimized module usage from poetry generation and downgrade PyTorch version to 2.3.1
ac829a5
abhisheksan
commited on