Requirements update (#851)

* Requirements update

* Code change to tricker workers

* update

Author: rasbt

pkg/llms_from_scratch/ch02.py

@@ -33,6 +33,7 @@ def __getitem__(self, idx):
 
 def create_dataloader_v1(txt, batch_size=4, max_length=256,
                          stride=128, shuffle=True, drop_last=True, num_workers=0):
+
     # Initialize the tokenizer
     tokenizer = tiktoken.get_encoding("gpt2")
 

requirements.txt

@@ -6,6 +6,6 @@ matplotlib >= 3.7.1        # ch04; ch06; ch07
 tensorflow>=2.16.2; sys_platform == 'darwin' and platform_machine == 'x86_64'  # Intel macOS
 tensorflow >= 2.18.0; sys_platform != 'darwin' or platform_machine != 'x86_64'   # ch05; ch06; ch07
 tqdm >= 4.66.1             # ch05; ch07
-numpy >= 1.26, < 2.1       # dependency of several other libraries like torch and pandas
+numpy >= 1.26             # dependency of several other libraries like torch and pandas
 pandas >= 2.2.1            # ch06
 psutil >= 5.9.5            # ch07; already installed automatically as dependency of torch