Fix issues with fake_llama

src/tests/conftest.py

src/tests/datasets/test_custom_dataset.py

src/tests/datasets/test_samsum_datasets.py

src/tests/test_batching.py

src/tests/test_chat_completion.py

src/tests/utils.py