Update README.md (#2)

- Update README.md (d4fe984209893c467c6ffcbac9de2d7bec7aa260)


Co-authored-by: みそかつおでん <[email protected]>

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)