Update README.md

README.md

@@ -4,21 +4,24 @@
 This repo contains the tokenizer, Chinese-Alpaca "merged" weights and configs for Chinese-LLaMA-Alpaca
 Directly load `merge` weight for chinese-llama-alpaca-plus-lora-7b
 
+```python
+model_name_or_path = 'we1kkk/chinese-llama-alpaca-plus-lora-7b'
 
-  model_name_or_path = 'we1kkk/chinese-llama-alpaca-plus-lora-7b'
-  
-  config = LlamaConfig.from_pretrained(
-      model_name_or_path,
-      # trust_remote_code=True
-  )
-  tokenizer = LlamaTokenizer.from_pretrained(
-      model_name_or_path,
-      # trust_remote_code=True
-  )
-  model = LlamaForCausalLM.from_pretrained(
-      model_name_or_path,
-      config=config,
-      ).half().cuda()
+config = LlamaConfig.from_pretrained(
+    model_name_or_path,
+    # trust_remote_code=True
+)
+tokenizer = LlamaTokenizer.from_pretrained(
+    model_name_or_path,
+    # trust_remote_code=True
+)
+model = LlamaForCausalLM.from_pretrained(
+    model_name_or_path,
+    config=config,
+    ).half().cuda()
+```
+
+ 
 
 ## Citation