Update README.md

README.md

@@ -40,6 +40,7 @@ model_id = "malhajar/phi-2-meditron"
 model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
                                              device_map="auto",
                                              torch_dtype=torch.float16,
+                                              trust_remote_code= True,
                                              revision="main")
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
@@ -52,7 +53,7 @@ prompt = '''
 
 ### Response:'''
 input_ids = tokenizer(prompt, return_tensors="pt").input_ids
-output = model.generate(inputs=input_ids,max_new_tokens=512,pad_token_id=tokenizer.eos_token_id,top_k=50, do_sample=True,trust_remote_code= True
+output = model.generate(inputs=input_ids,max_new_tokens=512,pad_token_id=tokenizer.eos_token_id,top_k=50, do_sample=True,
         top_p=0.95)
 response = tokenizer.decode(output[0])