Update README.md

We need remote code to run the example.

README.md

@@ -165,7 +165,8 @@ model_name_or_path = "TheBloke/Yarn-Llama-2-13B-128K-GPTQ"
 model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
                                              torch_dtype=torch.float16,
                                              device_map="auto",
-                                             revision="main")
+                                             revision="main",
+                                             trust_remote_code=True)
 
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)