diff --git a/ch04/01_main-chapter-code/previous_chapters.py b/ch04/01_main-chapter-code/previous_chapters.py index 7b4b4ea..b328743 100644 --- a/ch04/01_main-chapter-code/previous_chapters.py +++ b/ch04/01_main-chapter-code/previous_chapters.py @@ -27,7 +27,7 @@ class GPTDatasetV1(Dataset): return self.input_ids[idx], self.target_ids[idx] -def create_dataloader(txt, batch_size=4, max_length=256, stride=128, shuffle=True): +def create_dataloader_v1(txt, batch_size=4, max_length=256, stride=128, shuffle=True): # Initialize the tokenizer tokenizer = tiktoken.get_encoding("gpt2")