Update README.md

README.md

@@ -109,10 +109,11 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
-tokenizer = AutoTokenizer.from_pretrained("LLM360/CrystalChat", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("LLM360/CrystalChat", trust_remote_code=True).to(device)
+tokenizer = AutoTokenizer.from_pretrained("/mnt/ssd1/manaslu/checkpoints/hf/phase2a/CrystalCoder_phase2a_checkpoint_1600_to_hf", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("/mnt/ssd1/manaslu/checkpoints/hf/phase2a/CrystalCoder_phase2a_checkpoint_1600_to_hf", trust_remote_code=True).to(device)
+
+prompt = '<s> <|sys_start|> You are an AI assistant. You will be given a task. You must generate a detailed and long answer. <|sys_end|> <|im_start|> Write a python function that takes a list of integers and returns the squared sum of the list. <|im_end|>'
 
-prompt = 'int add(int x, int y) {'
 
 input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)
 gen_tokens = model.generate(input_ids, do_sample=True, max_length=400)
@@ -120,6 +121,28 @@ gen_tokens = model.generate(input_ids, do_sample=True, max_length=400)
 print("-"*20 + "Output for model"  + 20 * '-')
 print(tokenizer.batch_decode(gen_tokens)[0])
 ```
+
+Response:
+````
+Here's a Python function named `squared_sum_list` that takes a list of integers as input and returns the squared sum of the list:
+
+```python
+def squared_sum_list(lst):
+    return sum([num ** 2 for num in lst])
+```
+
+The function `squared_sum_list` uses a list comprehension to iterate over each number in the input list `lst` and calculate its square. Then, it uses the `sum` function to accumulate all the squared numbers in a single value - the squared sum of the list.
+
+For example:
+
+```python
+numbers = [1, 2, 3, 4, 5]
+print(squared_sum_list(numbers))  # Outputs: 55
+```
+
+In the above code, the list `[1, 2, 3, 4, 5]` is passed as an argument to the `squared_sum_list` function. The function calculates the sum of the squares of the elements in the list, which is `1 + 4 + 9 + 16 + 25 = 55`. The function then returns this result, which is printed to the console.</s>
+````
+
 <!-- ## CrystalChat DataMix
 | Subset      | Tokens (Billion) |
 | ----------- | ----------- |