Update README.md

README.md

@@ -109,8 +109,8 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
-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)
+tokenizer = AutoTokenizer.from_pretrained("LLM360/CrystalChat", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("LLM360/CrystalChat", 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|>'