Update README.md

README.md

@@ -23,6 +23,13 @@ import os
 import torch
 from transformers import AutoTokenizer, AutoConfig, AutoModel, BitsAndBytesConfig
 
+MODEL_ID = "./rednote-hilab/dots.llm1.inst"
+
+sys.path.append(os.path.abspath(MODEL_ID))
+
+from configuration_dots1 import Dots1Config
+from modeling_dots1 import Dots1ForCausalLM
+
 AutoConfig.register("dots1", Dots1Config)
 AutoModel.register(Dots1Config, Dots1ForCausalLM)