Update README.md

README.md

@@ -26,7 +26,7 @@ We use [AmberDateset](https://huggingface.co/datasets/LLM360/AmberDatasets) to t
 
 ## Result
 
-<img src=https://huggingface.co/LiqunMa/FBI-LLM_7B/resolve/main/main_result.jpg width="70%" />
+<img src=https://huggingface.co/LiqunMa/FBI-LLM_7B/resolve/main/main_result.jpg width="90%" />
 <!-- ![image](https://huggingface.co/LiqunMa/FBI-LLM_7B/resolve/main/main_result.jpg =200x) -->
 
 ## How to use
@@ -53,8 +53,7 @@ def load_model(model_size, model_dir):
     for p in ckpt_plist:
       weight_dict = torch.load(p)
       for k,v in _weight_dict.items():
-          if 'self_attn.rotary_emb.inv_freq' not in k:
-              weight_dict[k] = v
+          weight_dict[k] = v
 
     model.load_state_dict(weight_dict)
     for param in model.parameters():