Kit-Hung commited on
Commit
a466e3d
·
1 Parent(s): 53b8071

update: model path

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. download.py +7 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def init_models():
37
  #用初始化llm
38
  Settings.llm = llm
39
 
40
- documents = SimpleDirectoryReader("/home/user/llamaindex_demo/data").load_data()
41
  index = VectorStoreIndex.from_documents(documents)
42
  query_engine = index.as_query_engine()
43
 
 
37
  #用初始化llm
38
  Settings.llm = llm
39
 
40
+ documents = SimpleDirectoryReader("/home/user/data").load_data()
41
  index = VectorStoreIndex.from_documents(documents)
42
  query_engine = index.as_query_engine()
43
 
download.py CHANGED
@@ -4,6 +4,7 @@ def prepare_data():
4
  download_sentence_transformer()
5
  download_paraphrase_multilingual_MiniLM_L12_v2()
6
  download_NLTK()
 
7
 
8
 
9
  def download_sentence_transformer():
@@ -22,4 +23,9 @@ def download_paraphrase_multilingual_MiniLM_L12_v2():
22
  def download_NLTK():
23
  os.system('cd /home/user && git clone https://gitee.com/yzy0612/nltk_data.git --branch gh-pages')
24
  os.system('cd /home/user/nltk_data && mv /home/user/nltk_data/packages/* /home/user/nltk_data')
25
- os.system('cd /home/user/nltk_data/tokenizers && unzip punkt.zip && cd ./home/user/nltk_data/taggers && unzip averaged_perceptron_tagger.zip')
 
 
 
 
 
 
4
  download_sentence_transformer()
5
  download_paraphrase_multilingual_MiniLM_L12_v2()
6
  download_NLTK()
7
+ download_xtuner()
8
 
9
 
10
  def download_sentence_transformer():
 
23
  def download_NLTK():
24
  os.system('cd /home/user && git clone https://gitee.com/yzy0612/nltk_data.git --branch gh-pages')
25
  os.system('cd /home/user/nltk_data && mv /home/user/nltk_data/packages/* /home/user/nltk_data')
26
+ os.system('cd /home/user/nltk_data/tokenizers && unzip punkt.zip && cd ./home/user/nltk_data/taggers && unzip averaged_perceptron_tagger.zip')
27
+
28
+
29
+ def download_xtuner():
30
+ os.system('mkdir /home/user/data && git clone https://github.com/InternLM/xtuner.git')
31
+ os.system('mv /home/user/data/xtuner/README_zh-CN.md /home/user/data')